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;

}