POJ 2236 Wireless Network

题目链接

题意：距离d内的两台电脑互联。现在通过修理和查看操作，求查看时两台电脑是否互联？

分析：这是并查集的基本应用，两台修好的电脑若距离d内则加入合并。

#include<cstdio>
#include<algorithm>
#include<cstring>
#define MAX_N 1005
using namespace std;
struct node
{
    int x,y;
}computer[MAX_N];
int par[MAX_N];
int r[MAX_N];
int d,n;
bool repaired[MAX_N];
bool connected[MAX_N][MAX_N];
void init(int n)
{
    for(int i=0;i<=n;i++)
    {
        par[i]=i;
        r[i]=0;
        for(int j=i;j<=n;j++)
        {
            if((computer[i].x-computer[j].x)*(computer[i].x-computer[j].x)+(computer[i].y-computer[j].y)*(computer[i].y-computer[j].y)<=d*d)
            connected[i][j]=connected[j][i]=true;
            else
            connected[i][j]=connected[j][i]=false;
        }
    }
    return ;
}
int find(int x)
{
    if(x==par[x])
    return x;
    else
    return par[x]=find(par[x]);
}
void unite(int x,int y)
{
    x=find(x);
    y=find(y);
    if(x==y)
    return ;
    if(r[x]<r[y])
    par[x]=y;
    else
    {
        par[y]=x;
        if(r[x]==r[y])
        r[x]++;
    }
    return ;
}
bool same(int x,int y)
{
    return find(x)==find(y);
}
int main(void)
{
    char op;
    while(scanf("%d%d",&n,&d)==2)
    {
        memset(repaired,false,sizeof(repaired));
        for(int i=0;i<n;i++)
        scanf("%d%d",&computer[i].x,&computer[i].y);
        init(n);
        int p,q;
        while(scanf("%s",&op)==1)
        {
            if(op=='O')
            {
                scanf("%d",&p);
                p--;
                repaired[p]=true;
                for(int i=0;i<n;i++)
                {
                    if(i==p)
                    continue;
                    if(repaired[i]&&connected[i][p])
                    unite(p,i);
                }
            }
            else
            {
                scanf("%d%d",&p,&q);
                p--;
                q--;
                if(same(p,q))
                printf("SUCCESS\n");
                else
                printf("FAIL\n");
            }
        }
    }
    return 0;
}