POJ - 2236 Wireless Network（并查集）

POJ - 2236

题意：在平面坐标系上有n个坏掉的电脑，电脑之前只有距离小于d才能通信（在两台电脑都修好的情况下），

有两个查询第一种O x即修理电脑x，第二种S x y即查询x电脑和y电脑之间是否可以通信（间接的也可以）。

#include<iostream>
#include<cstdio>
using namespace std;
int f[1010],n,d;
struct note
{
	double x,y;
}q[1010];
char op;
bool p[1010];
void init() {for(int i=0;i<=1000;i++) f[i]=i;} 
int find(int x) {return f[x]==x?x:f[x]=find(f[x]);} 
void merge(int x,int y){f[find(x)]=find(y);} 
int main()
{
	scanf("%d%d",&n,&d);
	init();
	for(int i=1;i<=n;i++) scanf("%lf%lf",&q[i].x,&q[i].y);
	while(scanf(" %c",&op)!=EOF)
	{
		if(op=='O')
		{
			int k;
			scanf("%d",&k);
			p[k]=true;
			for(int i=1;i<=n;i++)
				if(p[i]&&i!=k&&d*d>=(q[i].x-q[k].x)*(q[i].x-q[k].x)+(q[i].y-q[k].y)*(q[i].y-q[k].y))
					merge(i,k);
		}
		else
		{
			int a,b;
			scanf("%d%d",&a,&b);
			if(find(a)==find(b)) puts("SUCCESS");
			else puts("FAIL"); 
		}
	}
	return 0;
}