Wireless Network POJ - 2236 -并查集

题意：一张图上分布着n台坏了的电脑，并知道它们的坐标。两台修好的电脑如果距离<=d就可以联网，也可以通过其他修好的电脑间接相连。给出操作“O x”表示修好x，给出操作“S x y”，请你判断x和y在此时有没有连接上。

思路：并查集

AC代码：

#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdlib>


using namespace std;
int dis;
int x[1002][1002];
int y[1002][1002];
int pre[20000],v[20000];
struct point
{
    int x;
    int y;


}p[1002];
int   distance(int a,int b,int c,int d)
{
     if(((a-c)*(a-c)+(b-d)*(b-d))<=(dis*dis))
        return true;
     else
        return false;
}
int find(int x)
{
    if(pre[x]==x)
        return x;
    else
    {
        int a=find(pre[x]);
        pre[x]=a;
        return pre[x];
    }
}
bool same(int a,int b)
{
    return find(a)==find(b);
}
void unite(int a,int b)
{
    int c=find(a);
    int d=find(b);
    if(c!=d)// 注意！！合并的是啥？父集！！
        pre[d]=c;
//    if(!same(a,b))//XXXXX错
//    {
//        pre[a]=b;
//        pre[b]=a;
//    }


}


int main()
{
   int n,a,b;
   cin>>n>>dis;


   for(int i=1;i<=n;i++)
   {
       scanf("%d%d",&p[i].x,&p[i].y);
   }
   for(int i=1;i<=n;i++)
   {
        pre[i]=i;
        v[i]=0;
   }


     char s;
   while(cin>>s)//此处scanf超时
{
        if(s=='O')
   {
       cin>>a;
       v[a]=1;
       for(int i=1;i<=n;i++)
       {
           if(i!=a&&v[i]&&distance(p[a].x,p[a].y,p[i].x,p[i].y))//注意！！要有v[i]!=0这个条件否则将没修好的电脑也算上了
              unite(a,i);


       }
   }
    else if(s=='S')
   {
       cin>>a>>b;
       if(same(a,b))
        cout<<"SUCCESS"<<endl;
       else
        cout<<"FAIL"<<endl;


   }


}




}