POJ 2236Wireless Network（并查集）

题意：

给定n个坏电脑的坐标，和最大距离d。如果两台电脑是被修好的，并且距离小于等于d，则这两台电脑是联通的。有两个指令： O+num，代表第num个电脑被修好    S+num1+num2，代表询问num1和num2是否联通，若是返回“ SUCCESS”，否则“FAIL”

做法：

用并查集做。不同的是，每次修好一个电脑，需要将这个电脑与所有被修好的电脑间遍历来建立联系，如果距离符合条件，则join这两个电脑。

AC代码：

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
int n,d;
int pre[1010];//用来记录父亲节点
int use[1010];//用来记录是否被修好 修好为1
struct Point //记录点坐标
{
    int x,y;
}point[1010];
bool isCommunicate(Point a,Point b)  //判定距离条件
{
    if((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)<=d*d)
        return true;
    else
        return false;
}
int find(int root)  //并查集的find函数，无压缩路径
{
    while(root!=pre[root])
    {
        root=pre[root];
    }
    return root;
}
void join(int a,int b)//并查集的join函数
{
    int roota,rootb;
    roota=find(a);
    rootb=find(b);
    if(roota!=rootb)
        pre[roota]=rootb;
}
int main()
{
    cin>>n>>d;
    memset(use,0,sizeof(use)); //初始化use
    for(int i=1;i<=n;i++) //记录每个point的数据
    {
        int xx,yy;
        cin>>xx>>yy;
        point[i].x=xx;
        point[i].y=yy;
        pre[i]=i;   //初始化 父亲节点即为本身
    }
    char ope;
    while(scanf("%c",&ope)!=EOF)
    {
        if(ope=='O')
        {
            int num;
            cin>>num;
            use[num]=1;
            for(int i=1;i<=n;i++)  //遍历所有修好的电脑
            {
                if(use[i]==1&&num!=i&&isCommunicate(point[num],point[i])) 
                {
                    join(num,i);
                }
            }
        }
        else if(ope=='S')
        {
            int num1,num2;
            cin>>num1>>num2;
            if(find(num1)==find(num2)) //如果两个电脑联通，输出
            {
                cout<<"SUCCESS"<<endl;
            }
            else
                cout<<"FAIL"<<endl;
        }
    }
    return 0;
}