poj 2236 Wireless Network 并查集

每修好一台电脑，检查是否可以与周围距离小于d的其他电脑合并为一个集合，能合并，表示能够相互通信。

s操作检查是否为同一个并查集。

/*
poj 2236
*/

#include <stdio.h>
#include <cmath>
#include <string.h>
#include <vector>
using namespace std;

#define maxn 2000
int N,d;
int p,q;
std::vector<int> dis[maxn];

int par[maxn];
int rank[maxn];

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]);
}

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]++;
	}

}

int main(int argc, char const *argv[])
{
	// freopen("input","r",stdin);

    int vis[maxn];
	memset(vis,false,sizeof(vis));

	scanf("%d%d",&N,&d);
	int x,y;
	pair<int,int> cood[maxn];
	for(int i=0;i<N;i++)
	{
		scanf("%d%d",&cood[i].first,&cood[i].second);
		for(int j=0;j<i;j++)
		{
			double t=sqrt( (cood[i].first-cood[j].first) * (cood[i].first-cood[j].first)
				+ (cood[i].second-cood[j].second) * (cood[i].second-cood[j].second) );

			if(t<=d)
			{
				dis[j].push_back(i);
				dis[i].push_back(j);
			}
		}
	}

	char c;

	init(N);



	while(getchar(),scanf("%c",&c)!=EOF)
	{
		if(c=='O')
		{
			scanf("%d",&p);
			p=p-1;
			vis[p]=1;
			for(int i=0;i<dis[p].size();i++)
			{
				if(vis[dis[p][i]])
				{
					unite(dis[p][i],p);
				}
			}
		}

		if(c=='S')
		{
			scanf("%d%d",&p,&q);
			p-=1;
			q-=1;
			// printf("%d %d\n",find(p),find(q) );

			if(find(p)==find(q))
				printf("SUCCESS\n");
			else
				printf("FAIL\n");
		}


	}

		return 0;
}