poj 2236 wireless network

这个题目很明显是并查集的题目.题目的意思就是现在有n台电脑,然后告诉你这些电脑的坐标,并且当两台电脑的距离小于或者等于这个给定的d时,这两台电脑视为是连通的,现在的问题就是输入S a b然后问你a和b是不是连通的.这个题目的大体的做法和通常的并查集是一样的,查找祖先,合并节点.但是在合并的时候首先要求是这个电脑必须被修复过,也就是下面的visited数组对应的值是1,并且这个距离是小于或者等于d.判断的时候只需要看一下这两个点的祖先是不是相同的就可以了.具体看下面的程序.

#include<iostream>
using namespace std;
int father[1005],visited[1005];
int find_ant(int x)
{
	if(x!=father[x])
		father[x]=find_ant(father[x]);
	return father[x];
}
void unin(int x,int y)
{
	int x_x,y_y;
	x_x=find_ant(x);
	y_y=find_ant(y);
	if(x_x!=y_y)
		father[x_x]=y_y;
}
struct node{
	int x;
	int y;
};
node ko[1005];
int map[1005][1005];
int main()
{
	int n,d,i,a,b,t1,t2,j;
	char c;
	cin>>n>>d;
	for(i=1;i<=n;i++)
	{
		father[i]=i;
		visited[i]=0;
		cin>>ko[i].x>>ko[i].y;
	}
	for(i=1;i<=n;i++)
		for(j=1;j<=n;j++)
			if((ko[i].x-ko[j].x)*(ko[i].x-ko[j].x)+(ko[i].y-ko[j].y)*(ko[i].y-ko[j].y)<=d*d)
				map[i][j]=1;
			else map[i][j]=0;
	while(cin>>c)
	{
		if(c=='S')
		{
			scanf("%d%d",&a,&b);
			t1=find_ant(a);
			t2=find_ant(b);
			if(t1==t2)
				cout<<"SUCCESS"<<endl;
			else cout<<"FAIL"<<endl;
		}
		if(c=='O')
		{
			scanf("%d",&a);
			for(j=1;j<=n;j++)
				if(visited[j]==1&&map[a][j]==1)
					unin(a,j);
			visited[a]=1;
		}
	}
	return 0;
}