【并查集】POJ2236-Wireless Network

【题目大意】

已知每一台电脑只能与它距离为d的电脑相连通，但是两台电脑间可以以第三台作为媒介连接。现在电脑全被损坏。每次可以进行两个操作中的一个，或是修好一台电脑，或是查询两台电脑是否连通。

【思路】

显然是并查集。每次修好一台新电脑，就与之前修好的每一台电脑进行判断，距离在d以内就合并。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=1001+50;
int n,d;
int x[MAXN],y[MAXN],rep[MAXN];
int h[MAXN],pa[MAXN];

int ind(int a,int b)
{
    return ((x[a]-x[b])*(x[a]-x[b])+(y[a]-y[b])*(y[a]-y[b])<=d*d);
}

void newset()
{
    for (int i=0;i<n;i++)
    {
        h[i]=0;
        pa[i]=i;
    }
}

int find(int x)
{
    int r=x;
    while (pa[r]!=r) r=pa[r];
    int p=x;
    while (pa[p]!=r)
    {
        int temp=pa[p];
        pa[p]=r;
        p=temp;
    }
    return r;
}

void unionset(int fa,int fb)
{
    if (h[fa]>=h[fb])
    {
        pa[fb]=fa;
        if (h[fa]==h[fb]) h[fa]++;
    }
    else
        pa[fa]=fb;
}

int main()
{
    scanf("%d%d",&n,&d);
    for (int i=0;i<n;i++)
    {
        scanf("%d%d",&x[i],&y[i]);
    }
    
    char c;
    int repd=0;//ÒѾ­Ð޺õĵçÄÔÊýÁ¿ 
    newset();
    
    getchar();
    while (scanf("%c",&c)!=EOF)
    {
        if (c=='S')
        {
            int a,b;
            scanf("%d%d",&a,&b);
            a--;
            b--;
            if (find(a)==find(b)) cout<<"SUCCESS"<<endl;else cout<<"FAIL"<<endl; 
        }
        
        else
        {
            int a;
            scanf("%d",&a);
            a--;
            for (int i=0;i<repd;i++)
            {
                if (ind(a,rep[i]))
                {
                    int fa=find(a),fb=find(rep[i]);
                    unionset(fa,fb); 
                }
            }
            rep[repd]=a;
            repd++;
        }
        getchar();
    }
    return 0;
}