POJ2236 wireless network 【并查集水题】

一张图上分布着n台坏了的电脑，并知道它们的坐标。两台修好的电脑如果距离<=d就可以联网，也可以通过其他修好的电脑间接相连。给出操作“O x”表示修好x，给出操作“S x y”，请你判断x和y在此时有没有连接上。

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
const int MAXN=111111;
const int MAX_N=111111;
int n,d;
using namespace std;
struct po
{
	int x,y;
	bool on;
}point[MAXN];
bool disok(int a,int b)
{
	int dx=point[a].x-point[b].x;
	int dy=point[a].y-point[b].y;
	if(dx*dx+dy*dy<=d*d)
		return true;
	else
		return false;
}
//并查集
int par[MAX_N];
int rank[MAX_N];//树的高度
void init(int n)
{
	for(int i=0;i<n;i++)
	{
		par[i]=i;
		rank[i]=0;
	}
}
int find(int x)//
{
		if(par[x]==x)
			return x;
		else
			return par[x]=find(par[x]);
}
//合并x和y所属的集合
void unite(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x==y)
		return;
	if(rank[x]<rank[y])
		par[x]=y;
	else
	{
		par[y]=x;
		if(rank[x]==rank[y])
			rank[x]++;
	}
}
//判断x和y是否同属于一个集合
bool same(int x, int y)
{
	return find(x)==find(y);
}
int main()
{
	#ifndef ONLINE_JUDGE
		freopen("G:/1.txt","r",stdin);
		freopen("G:/2.txt","w",stdout);
	#endif
	scanf("%d%d",&n,&d);
	init(n);
	for(int i=1;i<=n;i++)
		scanf("%d%d",&point[i].x,&point[i].y);
	char kind[2];
	while(scanf("%s",kind)!=EOF)
	{
		if(strcmp(kind,"O")==0)
		{
			int nn;
			scanf("%d",&nn);
			point[nn].on=true;
			for(int i=1;i<=n;i++)
			{
				if(point[i].on&&disok(i,nn))
					unite(nn,i);
			}
		}
		else
		{
			int s1,s2;
			scanf("%d%d",&s1,&s2);
			if(same(s1,s2))
			{
				cout<<"SUCCESS"<<'\n';
			}
			else
			{
				cout<<"FAIL"<<'\n';
			}
		}
	}
}