POJ2236 Wireless Network 并查集

Wireless Network

题目链接：http://poj.org/problem?id=2236
这道题是一个简单的并查集，唯一不同的是，一开始所有点都是不可用的，要进行一个操作才能将点维修好，还有一个就是查询，基本没啥改变，就是一个并查集板子题吧。
另外要提到的就是路径压缩的问题，虽然这道题并没有这个要求，但在其他的题目上可能会需要压缩路径来节约许多查找的时间

如下：

int findrt(int x)    //递归查找压缩
{

    if(fab[x]==x)
        return x;
    return fab[x] = findrt(fab[x]);
}

int findrt(int x)   //无压缩循环查找
{
    while(fab[x]!=x)
        x = fab[x];
    return x;
}

代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <math.h>
using namespace std;
int fab[1005];
int flag[1005];
typedef struct
{
    int x,y;
}node;
int findrt(int x)
{

    if(fab[x]==x)
        return x;
    return fab[x] = findrt(fab[x]);
}
void Merge(int a,int b)
{
    int fa = findrt(a);
    int fb = findrt(b);
    if(fa!=fb)
        fab[fa] = fb;
}
int main()
{
    memset(flag,0,sizeof(flag));
    int d;
    int n,i,j,h,x,y;
    char c[10];
    node point[1005];
    scanf("%d %d",&n,&d);
    for(i=1;i<=n;i++)
    {
        scanf("%d %d",&point[i].x,&point[i].y);
        fab[i] = i;
    }
    while(scanf("%s",c)!=EOF)
    {
        if(c[0]=='S')
        {
            scanf("%d %d",&x,&y);
            if(findrt(x)==findrt(y))
                printf("SUCCESS\n");
            else
                printf("FAIL\n");
        }
        else
        {
            scanf("%d",&h);
            if(flag[h]==1)continue;
            flag[h] = 1;
            for(i=1;i<=n;i++)
            {
                if((point[i].x-point[h].x)*(point[i].x-point[h].x)+(point[i].y-point[h].y)*(point[i].y-point[h].y)<=d*d&&flag[i]!=0)
                    Merge(i,h);
            }
        }
    }
    return 0;
}