POJ 2236 Wireless Network 简单并查集

题目链接：http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66964#problem/A

题意：由于地震计算机系统坏了，现在，逐个修复某些电脑，并给出一些询问，问两个计算机是否可以通信。

虽然题目简单，但是能够一次AC感觉已经不错了，

代码：

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;

int Map[1010][1010],X[1010],Y[1010];
int vis[1010],pre[1010];
int N,d;

int find(int x){
    int t=x;
    while(pre[t]!=t) t=pre[t];
    while(x!=t) pre[x]=t,x=pre[x];
    return t;
}

void join(int t){
    vis[t]=1;
    for(int i=1;i<=N;i++) if(i!=t && vis[i] && Map[i][t])
        pre[find(t)]=find(i);
}

void judge(int x,int y){
    if(find(x)==find(y)) printf("SUCCESS\n");
    else printf("FAIL\n");
}

int main(){
    //freopen("D:\\in.txt","r",stdin);
    memset(Map,0,sizeof(Map));//表示两个点是否在d范围内
    memset(vis,0,sizeof(vis));//这个点是否已经修复
    cin>>N>>d;
    for(int i=0;i<=N;i++) pre[i]=i;
    for(int i=1;i<=N;i++) scanf("%d %d",&X[i],&Y[i]);
    for(int i=1;i<=N;i++) for(int j=1;j<=N;j++)
        if((X[i]-X[j])*(X[i]-X[j])+(Y[i]-Y[j])*(Y[i]-Y[j])<=d*d)
            Map[i][j]=Map[j][i]=1;
    char ch;getchar();
    while(scanf("%c",&ch)==1){
        if(ch=='O'){
            int t;scanf("%d",&t);
            join(t);
        }else{
            int a,b;
            scanf("%d %d",&a,&b);
            judge(a,b);
        }
        getchar();
    }
    return 0;
}