POJ——2236 Wireless Network（并查集模板题）

原题链接： http://poj.org/problem?id=2236

测试样例

Sample Input
4 1
0 1
0 2
0 3
0 4
O 1
O 2
O 4
S 1 4
O 3
S 1 4
Sample Output
FAIL
SUCCESS

题意： 有$n$台处于平面的电脑都坏了。已知两台电脑的直接通信距离为$d$（即若两台已经修复好的电脑它们的直接距离小于等于$d$，则可以直接通信。）。现在进行一系列的操作：若为 $Ox$，则此操作是对编号为$x$的电脑进行修复。若为$Sxy$，则是测试编号为$x$和编号为$y$的电脑能否通信，需要你输出答案。

解题思路： 一道很有趣的并查集，为什么呢？因为合并操作是要我们自己去进行的。系统每进行一次$O$操作，我们都要对所有的计算机进行判断它们是否能通信。是否能通信也就是是否处于同一集合，这里即是模板并查集。重要的一点就是在于我们怎么去合并？我们要引入一个$vis$数组来判断电脑是否已经被修复，若没有被修复自然是无法合并，我们还要判断两台电脑之间的距离是否小于等于$d$。这两个限制条件即决定了能否通信。 知道了这个，则此题易解。具体看代码。

AC代码

/*
*邮箱：[email protected]
*blog:https://me.csdn.net/hzf0701
*注：文章若有任何问题请私信我或评论区留言，谢谢支持。
*
*/
#include<iostream>	//POJ不支持

#define rep(i,a,n) for (int i=a;i<=n;i++)//i为循环变量，a为初始值，n为界限值，递增
#define per(i,a,n) for (int i=a;i>=n;i--)//i为循环变量， a为初始值，n为界限值，递减。
#define pb push_back
#define IOS ios::sync_with_stdio(false);cin.tie(0); cout.tie(0)
#define fi first
#define se second
#define mp make_pair

using namespace std;

const int inf = 0x3f3f3f3f;//无穷大
const int maxn = 1e3+4;//最大值。
typedef long long ll;
typedef long double ld;
typedef pair<ll, ll>  pll;
typedef pair<int, int> pii;
//*******************************分割线，以上为自定义代码模板***************************************//

int n,d;//每台电脑所能通信的最大距离。
bool vis[maxn];//判断电脑是否修好。
int point[maxn][2];//存储每台电脑的坐标
int father[maxn];//存储电脑所在集合。
int Find(int x){
    
    
	int r=x;
	while(r!=father[r]){
    
    
		r=father[r];
	}
	int i=x,j;
	while(father[i]!=r){
    
    
		j=father[i];
		father[i]=r;
		i=j;
	}
	return r;
}
int main(){
    
    
	//freopen("in.txt", "r", stdin);//提交的时候要注释掉
	IOS;
	while(cin>>n>>d){
    
    
		rep(i,1,n){
    
    
			cin>>point[i][0]>>point[i][1];
			father[i]=i;
			vis[i]=false;
		}
		string op;
		int temp1,temp2;
		while(cin>>op){
    
    
			if(op=="O"){
    
    
				cin>>temp1;
				vis[temp1]=true;
				rep(i,1,n){
    
    
					if(i!=temp1&&vis[i]){
    
    
						int dx=point[temp1][0]-point[i][0];
						int dy=point[temp1][1]-point[i][1];
						if(dx*dx+dy*dy<=d*d){
    
    
							int fx=Find(i);
							int fy=Find(temp1);
							if(fx!=fy){
    
    
								father[fx]=fy;
							}
						}
					}
				}
			}
			else{
    
    
				cin>>temp1>>temp2;
				if(Find(temp1)!=Find(temp2)){
    
    
					cout<<"FAIL"<<endl;
				}
				else{
    
    
					cout<<"SUCCESS"<<endl;
				}
			}
		}
	}
	return 0;
}