【并查集】POJ2236-Wireless Network

【题目大意】

已知每一台电脑只能与它距离为d的电脑相连通，但是两台电脑间可以以第三台作为媒介连接。现在电脑全被损坏。每次可以进行两个操作中的一个，或是修好一台电脑，或是查询两台电脑是否连通。

【思路】

显然是并查集。每次修好一台新电脑，就与之前修好的每一台电脑进行判断，距离在d以内就合并。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 using namespace std;
 const int MAXN=+;
 int n,d;
 int x[MAXN],y[MAXN],rep[MAXN];
 int h[MAXN],pa[MAXN];

 int ind(int a,int b)
 {
     return ((x[a]-x[b])*(x[a]-x[b])+(y[a]-y[b])*(y[a]-y[b])<=d*d);
 }

 void newset()
 {
     for (int i=;i<n;i++)
     {
         h[i]=;
         pa[i]=i;
     }
 }

 int find(int x)
 {
     int r=x;
     while (pa[r]!=r) r=pa[r];
     int p=x;
     while (pa[p]!=r)
     {
         int temp=pa[p];
         pa[p]=r;
         p=temp;
     }
     return r;
 }

 void unionset(int fa,int fb)
 {
     if (h[fa]>=h[fb])
     {
         pa[fb]=fa;
         if (h[fa]==h[fb]) h[fa]++;
     }
     else
         pa[fa]=fb;
 }

 int main()
 {
     scanf("%d%d",&n,&d);
     for (int i=;i<n;i++)
     {
         scanf("%d%d",&x[i],&y[i]);
     }

     char c;
     int repd=;//ÒѾ­Ð޺õĵçÄÔÊýÁ¿
     newset();

     getchar();
     while (scanf("%c",&c)!=EOF)
     {
         if (c=='S')
         {
             int a,b;
             scanf("%d%d",&a,&b);
             a--;
             b--;
             if (find(a)==find(b)) cout<<"SUCCESS"<<endl;else cout<<"FAIL"<<endl;
         }

         else
         {
             int a;
             scanf("%d",&a);
             a--;
             for (int i=;i<repd;i++)
             {
                 if (ind(a,rep[i]))
                 {
                     int fa=find(a),fb=find(rep[i]);
                     unionset(fa,fb);
                 }
             }
             rep[repd]=a;
             repd++;
         }
         getchar();
     }
     return ;
 }