poj2236 并查集

 

 

   如题：http://poj.org/problem?id=2236

  

    修复一个网络。输入每台电脑的x，y。

     0 X代表修复编号X的计算机。S X Y代表判断X Y这2台计算机能否通信。

 

 

   唯一一点需要注意的是在并入集合的时候，虽然修复了，不一定能通信。加一个repair域，只要O x，计算机x的repair域就是1。以后修复时，再来判断有没有可能进行通信。

 

  

#include<cstdio>
#include<cmath>

#define N 1005

typedef struct C{ 
 int x, y;
 int repair;
}C;

C Comp[N];

int f[N];
int n,d;  //电脑数量，最大长度

int find(int x)  //找集合
{
 if(x==f[x])
  return x;
 else return f[x]=find(f[x]);
}

double distance(C  a,C  b) //返回2台电脑之间的距离
{
 return sqrt((double)(a.x-b.x)*(double)(a.x-b.x)+(double)(a.y-b.y)*(double)(a.y-b.y));
}
 
void Union(int x)  //虽然被修了，但不一定能通信，这里只要修就让repair域置1，
      //如果不能通信，不并入集合，等待并入其他元素时，和能通信的并入一个集合。
{
 int i;
 Comp[x].repair=1;
 
 for(i=1;i<=n;i++)
  if(Comp[i].repair==1&&distance(Comp[i],Comp[x])<d+1.0e-8)
   f[find(i)]=x;
}
int main()
{
 //freopen("C:\\1.txt","r",stdin);
  char ch;
  int i;
  scanf("%d%d",&n,&d);
  for(i=1;i<=n;i++)
  {
   scanf("%d%d",&Comp[i].x,&Comp[i].y); 
   f[i]=i;  //初始化集合
  }
  getchar();
  
  while(~scanf("%c",&ch))
  {
   if(ch=='O')   //修复电脑，合并
   {
    int num;
   scanf("%d",&num);
   getchar();
    Union(num);  //并入集合
    continue;
   }
   if(ch=='S')
   {
    int x,y;
    scanf("%d%d",&x,&y);
    getchar();
    if(find(x)==find(y))  //如果属于一个集合
     printf("SUCCESS\n");
    else
     printf("FAIL\n");
   }
  }
  return 0;
}