POJ - 2236 Wireless Network （并查集）

题目：http://poj.org/problem?id=2236

题意：n台损坏的计算机分布在二维平面的一些点上，给出n台计算机的位置，给出两种操作，一是修复某台计算机，二是 查询两台计算机之间是否能联络。

分析：好久没做并查集了，一开始想着用深搜，尴尬，后来分析搜索明显超时，明显一道裸的并查集。两台计算机能联络必须满足两个条件，一是距离d’<=d，二是两台计算机完好。（用个标记）并查集的本质是一些有联系的点合并为一个集合，这个集合有唯一的公共祖先。（debug了几天，看来对并查集的本质理解不深刻）

代码：

#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
const int maxn = 1e3+10;
struct node{
int x,y,flag;
}node[maxn];
int pre[maxn],n,d,s,e;

void init(){
for(int i=1;i<=n;pre[i]=i,i++);
}

int Find(int x){
while(x!=pre[x])
x=pre[x];
return x;
}

int main(){
cin>>n>>d;
for(int i=1;i<=n;i++){
cin>>node[i].x>>node[i].y;
node[i].flag=0;
}
char c;
init();
while(cin>>c){
if(c=='O'){
cin>>s;
node[s].flag=1;
for(int i=1;i<=n;i++){
if(i==s) continue;
if(Find(i)!=Find(s)&&node[i].flag&&d*d>=((node[i].x-node[s].x)*(node[i].x-node[s].x)+(node[i].y-node[s].y)*(node[i].y-node[s].y)))
pre[Find(i)]=Find(s);
}
}
else if(c=='S'){
cin>>s>>e;
if(Find(s)!=Find(e)) cout<<"FAIL"<<endl;
else cout<<"SUCCESS"<<endl;
}
}
}