PKU 2236 Wireless Network（并查集）

 

http://acm.pku.edu.cn/JudgeOnline/problem?id=2236

 

题目大意：亚洲海底地震，使得所有的电脑都坏了并中断了联系，现在有一群维修人员被派往现场修复。他们在那边会做两件事，1.“O”代表要修复某编号的电脑，“S”代表要测试两台编号不同的电脑是否可以连通，测试成功，则输出成功，否则失败。但是两台电脑之间如果要直接连通的话，它们之间的距离不能够超过OJ提供的距离D，如果超过的话，他们要连通的话，中间就要有中介电脑才行。现在你的工作就是根据修复人员所做的操作，输出成功或者失败。

 

解题思路：这道的话就是属于互通的题目啦，也就是要用到并查集。修复一台电脑就标记为修复过了，然后跟其他已经修复过的电脑进行合并，但是合并的条件必须是两台电脑的距离小于D。当要测试两台电脑时，就是判断他们的根结点是否一样或者是距离小于D，然后再进行合并。

 

#include <iostream> #include <math.h> using namespace std; #define size 1002 int parents[size]; int deep[size]; bool visited[size]; struct coordinate { double x; double y; }locate[size]; void CreateSet(int x)//创建集合 { parents[x] = x; deep[x] = 1; visited[x] = false; } int Find(int x)//寻找根结点 { int r = x; int temp; while(r!=parents[r]) r = parents[r]; while (x!=r) { temp = parents[x]; parents[temp] = r; x = temp; } return r; } void Union(int a,int b)//合并集合 { int t1 = Find(a); int t2 = Find(b); if(deep[t1]>deep[t2]) parents[t2] = t1; else parents[t1] = t2; if(deep[t1] == deep[t2]) deep[t2]++; } bool IsOk(int a,int b,long distance)//判断两台电脑是否可以连通 { double k; k = sqrt((locate[a].x-locate[b].x)*(locate[a].x-locate[b].x)+(locate[a].y-locate[b].y)*(locate[a].y-locate[b].y)); if(k<=distance) return true; else return false; } int main() { char commander; int computer; int p1,p2; int i; long distance; cin>>computer>>distance; for(i=1;i<=computer;i++) CreateSet(i); for (i=1;i<=computer;i++) cin>>locate[i].x>>locate[i].y; while (cin>>commander) { if (commander == 'O') { cin>>p1; visited[p1] = true; /*查看修复的电脑是否可以加入到并查集中*/ for(i=1;i<=computer;i++) { if(visited[i]&&i!=p1) if(IsOk(p1,i,distance)) if(Find(p1)!=Find(i)) Union(p1,i); } } else { cin>>p1>>p2; if(IsOk(p1,p2,distance)||Find(p1)==Find(p2)) { cout<<"SUCCESS"<<endl; if(Find(p1)!=Find(p2)) Union(p1,p2); } else cout<<"FAIL"<<endl; } } return 0; }