[BZOJ2429][HAOI2006]聪明的猴子(最小生成树)

性质：最小生成树上任意两点间的最大边权，一定是这两点间所有路径的最大边权中最小的。证明显然。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define rep(i,l,r) for (int i=(l); i<=(r); i++)
using namespace std;

const int N=1010,M=1000010;
int n,m,tot,ans,mx,k,a
,fa
;
struct P{ int x,y; }p
;
struct E{ int u,v,w; }e[M];
bool operator <(const E &a,const E &b){ return a.w<b.w; }
int sqr(int x){ return x*x; }
int get(int x){ return (fa[x]==x) ? x : fa[x]=get(fa[x]); }

int main(){
scanf("%d",&m);
rep(i,1,m) scanf("%d",&a[i]);
scanf("%d",&n);
rep(i,1,n) scanf("%d%d",&p[i].x,&p[i].y),fa[i]=i;
rep(i,1,n-1) rep(j,i+1,n) e[++tot]=(E){i,j,sqr(p[i].x-p[j].x)+sqr(p[i].y-p[j].y)};
sort(e+1,e+tot+1);
rep(i,1,tot){
int u=e[i].u,v=e[i].v;
if (get(u)==get(v)) continue;
k++; mx=max(mx,e[i].w);
if (k==n-1) break;
fa[get(u)]=get(v);
}
rep(i,1,m) if (sqr(a[i])>=mx) ans++;
printf("%d\n",ans);
return 0;
}