Wireless Network（并查集）

POJ - 2236

 #include<iostream>
 #include<algorithm>
 #include<cstring>
 #include<cmath>
 using namespace std;

 int n,d;
 char c;
 int p,q;
 int book[]={},f[]; //分别是 标记是否修好 ， 找爸 

 struct note
 {
     int x,y;
 }a[]; //记录坐标 

 int getf(int u)
 {
     return u==f[u]? u:getf(f[u]);
 }

 void merge(int x,int y)
 {
     x=getf(x);
     y=getf(y);
     if(x!=y)
     {
         f[y]=x;
     }
 }

 int main()
 {
     cin>>n>>d;
     for(int i=;i<=n;i++)
     {
         f[i]=i;
         cin>>a[i].x>>a[i].y;
     }
     while(cin>>c)
     {
         if(c=='O') //修
         {
             cin>>p;
             book[p]=;  //修好了 ,下面进行寻找
             for(int i=;i<=n;i++)
             {
                 //第i个电脑不和第p的电脑超出距离，并且第i个电脑已经被修好 是能连通的基本条件
                 if( d*d >= 1.0*((a[p].x-a[i].x)*(a[p].x-a[i].x)+(a[p].y-a[i].y)*(a[p].y-a[i].y)) && book[i]==)
                     merge(f[p],f[i]); //这里左右位置不要变，要一直都把电脑p作为爸爸
             }
         }
         else if(c=='S')
         {
             cin>>p>>q;
             if(getf(p)==getf(q))
                 cout<<"SUCCESS"<<endl;
             else
                 cout<<"FAIL"<<endl;
         }
     }
 }