【 POJ - 2236】 Wireless Network （并查集）

描述

所有电脑都是坏的
工人一次修复一台电脑

电脑之间的通信有两种：
距离<=d 直接通信
距离>d 通过其他好电脑间接通信

判断某两台电脑能不能通信

解决

我们把距离<=d 看成是有边相连
将电脑修好 看成合并并查集
判断两台电脑能不能通信就变成是否在同一个并查集内

代码

#include<cstdio>
#include<vector>
typedef long long LL;
const int N=1e3+10; 
const int oo=0x3f3f3f3f; 
using namespace std;
bool flag;
int fa[N],x[N],y[N];
int read(){
	int f=1,s=0;char c=getchar();
	for(;c<'0'||c>'9';c=getchar())if(c=='-')f=-1;
	for(;c>='0'&&c<='9';c=getchar())s=s*10+c-48;
	return f*s;
}

int find(int x){
	if(fa[x]!=x)fa[x]=find(fa[x]);
	return fa[x]; 
}
void join(int x,int y){
	x=find(x);y=find(y);
	if(x==y)return;
	fa[x]=y;
}
vector<int>a[N];
bool b[N];
int sqr(int x){return x*x;}
int main(){
	int n=read(),d=read();
	for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=1;i<=n;i++)
		x[i]=read(),y[i]=read();
	for(int i=1;i<=n;i++)
		for(int j=i+1;j<=n;j++)
			if(sqr(x[i]-x[j])+sqr(y[i]-y[j])<=sqr(d))
				a[i].push_back(j),a[j].push_back(i);
	char c;int p,q;
	while(scanf("%c",&c)!=EOF){
		switch(c){
			case 'O':{
				p=read();
				b[p]=true;
				for(int i=0;i<a[p].size();i++)
					if(b[a[p][i]])join(p,a[p][i]);
				break;
			}
			case 'S':{
				p=find(read());q=find(read());
				printf("%s\n",p==q?"SUCCESS":"FAIL");
				break;
			}
		}
	}
	return 0;
}