BZOJ 2429 [HAOI2006]聪明的猴子

【题解】

　　先跑一边最小生成树，记录树上最大的边权，然后对于每只猴子判断其跳跃距离是否大于等于最大边权即可。同时为了避免小数的麻烦，边权保留勾股定理后的平方结果，把猴子的跳跃距离平方一下，因为都是非负数，平方一下不改变大小关系。

　　

#include<cstdio>
#include<algorithm>
#define N 1010
#define rg register
using namespace std;
int n,m,mx,tot,ans,cnt,dis[N],fa[N];
struct rec{
    int u,v,dis;
}a[N*N];
struct point{
    int x,y;
}p[N];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
}
inline bool cmp(rec a,rec b){return a.dis<b.dis;}
inline int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
int main(){
    m=read();
    for(rg int i=1;i<=m;i++) dis[i]=read(),dis[i]*=dis[i];
    n=read();
    for(rg int i=1;i<=n;i++){
        p[i].x=read(),p[i].y=read();
    }
    for(rg int i=1;i<n;i++)
        for(rg int j=i+1;j<=n;j++)
        a[++tot]=(rec){i,j,(p[i].x-p[j].x)*(p[i].x-p[j].x)+(p[i].y-p[j].y)*(p[i].y-p[j].y)};
    sort(a+1,a+1+tot,cmp);
    for(rg int i=1;i<=n;i++) fa[i]=i;
    for(rg int i=1;i<=tot;i++){
        int u=find(a[i].u),v=find(a[i].v);
        if(u!=v) mx=max(mx,a[i].dis),fa[u]=v,cnt++;
        if(cnt>=n-1) break;    
    }
    for(rg int i=1;i<=m;i++) if(dis[i]>=mx) ans++;
    printf("%d\n",ans);
    return 0;
}