POJ-2236（并查集）

Wireless NetWork

POJ-2236

• 需要注意这里的树的深度需要初始化为0。
• 而且，find函数需要使用路径压缩，这里的unint合并函数也使用了优化（用一开始简单的合并过不了）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<cmath>
using namespace std;
int n,d;
bool repaired[1002];
struct node{
    int x;
    int y;
};
node com[1002];
int set[1002];
int rank1[1002];//rank1[i]表示i的深度
int find(int x){
    if(x==set[x])
        return set[x];
    return set[x]=find(set[x]);
}
void unint(int a,int b){
    int ta=find(a);
    int tb=find(b);
    if(ta!=tb){
        if(rank1[ta]<rank1[tb]){
            set[ta]=tb;
        }else{
            set[tb]=ta;
            if(rank1[ta]==rank1[tb]){
                rank1[ta]++;
            }
        }
    }else return;
}
double compute(int a,int b){
    return (com[a].x-com[b].x)*(com[a].x-com[b].x)+(com[a].y-com[b].y)*(com[a].y-com[b].y);
}
int main(){
    for(int i=0;i<1002;i++){
        set[i]=i;
        rank1[i]=0;
    }
    cin>>n>>d;
    int x,y;
    for(int i=1;i<=n;i++){
        scanf("%d%d",&com[i].x,&com[i].y);
    }
    char c;
    while(scanf("%c",&c)!=EOF){
        if(c=='O'){//修复
            int num;
            scanf("%d",&num);
            repaired[num]=true;
            for(int i=1;i<=n;i++){
                if(i!=num&&repaired[i]&&compute(num,i)<=d*d){
                    unint(i,num);
                }
            }
        }else if(c=='S'){
            int x,y;
            scanf("%d%d",&x,&y);
            if(repaired[x]&&repaired[y]){
                int t1=find(x);
                int t2=find(y);
                if(t1==t2){
                    cout<<"SUCCESS"<<endl;
                }else cout<<"FAIL"<<endl;
            }else{
                cout<<"FAIL"<<endl;
            }
        }
    }
    return 0;
}