POJ2236 Wireless Network 并查集

题目大意：输入n和d，n代表标号为1到n的电脑，下面n行每行是一台电脑的坐标，每两台电脑只有相距不超过d时才算相连，然后给出若干组操作，每一组操作表示维修电脑x或者询问x和y是否相连，让判断对于每次询问的结果。

用并查集纪录每一次维修后可以相连的电脑，然后对于每一次询问，我们只需判断x和y是否有相同的根结点即可。

#include <cstdio>
#include <iostream>
#include <cmath>
using namespace std;
#define MAX 1010
int par[MAX],rank[MAX],s[MAX];
typedef struct node
{
    int x,y;
}Point;
Point p[MAX];
void Init(int n)
{
    for(int i=1;i<=n;i++)
    {
        par[i]=i;
        rank[i]=1;
    }
}
int Find(int x)
{
    if(par[x]!=x)
       par[x]=Find(par[x]);
    return par[x];
}
void Union(int x,int y)
{
    x=Find(x);
    y=Find(y);
    if(x==y) return ;
    if(rank[x]>rank[y])
    {
        par[y]=x;
        rank[x]+=rank[y];
    }
    else
    {
        par[x]=y;
        rank[y]+=rank[x];
    }
}
int Dist(Point a,Point b)
{
    return (a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y);
}
int main()
{
    int n,d,i,ans,cnt;
    char tmp[2];
    scanf("%d%d",&n,&d);
    Init(n);
    cnt=0;//cnt表示已维修的电脑的数目
    for(i=1;i<=n;i++)
      scanf("%d%d",&p[i].x,&p[i].y);
    while(scanf("%s",tmp)!=-1)
    {
        if(tmp[0]=='O')
        {
            scanf("%d",&ans);//每一次有维修出现时，我们都拿它和之前已维修的可以直接相连的电脑合并
            for(i=0;i<cnt;i++)
              if(Dist(p[s[i]],p[ans])<=d*d)
                Union(s[i],ans);
              s[cnt++]=ans;
        }
        else
        {
            int x,y;
            scanf("%d%d",&x,&y);
            if(Find(x)==Find(y))
              puts("SUCCESS");
            else puts("FAIL");
        }
    }
    return 0;
}