poj 2236 并查集 计算机网络

题意：给出N，d代表有N个计算机，还有可以建立连接的限制距离d。O 后面的数字代表了维修好的计算机，S 后面的两个数字用来判断，这两个计算机是否建立的连接。

思路：并查集，每修好一个计算机，进行标记，同时枚举所有计算机，在维修好的计算机中，找出可以建立连接的计算机，进行合并。

#include<iostream>
#include<cmath>
using namespace std;
struct point
{
  int x;
  int y;
  int parent;
}p[1010];
bool vis[1010];
int N,d;
double get_dis(int i,int j)//得到距离
{
  int x1,y1,x2,y2;
  x1=p[i].x;y1=p[i].y;
  x2=p[j].x;y2=p[j].y;
  return sqrt(double((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)));
}
void make_set()
{
  for(int i=1;i<=N;i++)
  p[i].parent=i;
}
int find_set(int x)
{
  if(x==p[x].parent)  return x;
  else
  {
    p[x].parent=find_set(p[x].parent);
  }
  return p[x].parent;
}
void join(int x,int y)
{
  int r1=find_set(x);
  int r2=find_set(y);
  if(r1!=r2)
  {
    p[r1].parent=r2;
  }
}
int main()
{
  char s[2];
  int g,a,b;
  scanf("%d%d",&N,&d);
  for(int i=1;i<=N;i++)
  {
    scanf("%d%d",&p[i].x,&p[i].y);
    vis[i]=0;
  }
  make_set();
  //for(int i=1;i<=N;i++)
  //printf("%d,%d\n",p[i].x,p[i].y);
  while(scanf("%s",s)!=EOF)
  {
    if(s[0]=='O')
    {
      scanf("%d",&g);
      vis[g]=1;
      for(int i=1;i<=N;i++)//枚举所有计算机
      {
        if(vis[i]&&i!=g)//对已经修好的计算机，判断是否可以建立连接
        {
          //printf("%d %d %f\n",i,g,get_dis(g,i));
          //system("pause");
          if(get_dis(g,i)<=d)
          join(g,i);
        }
      }
    }
    else
    {
      scanf("%d%d",&a,&b);
      if(find_set(a)==find_set(b))
      printf("SUCCESS\n");
      else
      printf("FAIL\n");
    }
  }
}