poj 2236Wireless Network (并查集的应用）

题目的意思就是现在有n台电脑,然后告诉你这些电脑的坐标,并且当两台电脑的距离小于或者等于

这个给定的d时,这两台电脑视为是连通的,现在的问题就是输入S a b然后问你a和b是不是连通的.

这个题目的大体的做法和通常的并查集是一样的,查找祖先,合并节点.但是在合并的时候首先要求

是这个电脑必须被修复过,也就是下面的visited数组对应的值是1,并且这个距离是小于或者等于d.

判断的时候只需要看一下这两个点的祖先是不是相同的就可以了.

能够合并在一起的两个条件：

满足距离的要求和已经修复；

#include"stdio.h"

#include"string.h"
int pre[2000];
int mark[2000];
int a[10000],b[10000];
int find(int k)
{
if(k!=pre[k])
pre[k]=find(pre[k]);
return pre[k];
}
void Union(int i,int j)
{
int xx,yy;
xx=find(i);
yy=find(j);
if(xx!=yy)
pre[yy]=xx;
}
int main()
{
int i,k,h,m,n;
char str[100];
    scanf("%d%d",&m,&n);
for(i=0;i<=m;i++)
pre[i]=i;
memset(mark,0,sizeof(mark));
for(i=1;i<=m;i++)
scanf("%d%d",&a[i],&b[i]);
while(scanf("%s",str)!=EOF)
{
if(str[0]=='O')
{
scanf("%d",&k);
mark[k]=1;
for(i=1;i<=m;i++)
{
if(((a[i]-a[k])*(a[i]-a[k])+(b[i]-b[k])*(b[i]-b[k]))<=n*n&&mark[i])
{
Union(k,i);
}
}
}
else
{
scanf("%d%d",&k,&h);
if(find(k)==find(h)&&mark[k]&&mark[h])
printf("SUCCESS\n");
else
printf("FAIL\n");
}
}
return 0;
}