POJ--2236--Wireless Network

并查集的简单应用

由于只有1001个点，所有可以预处理每个点到其他任一点的距离。

然后对于修复操作来说。假设修复的点是x。

则：

依次检查每一个点ｉ，如果点ｉ已经修复了且距离小于等于ｄ，则合并点ｉ和点ａ的集合

对于询问来说，直接查找就可以了

#include<iostream>
#include<cstring>
#include<cstdio>
#define maxn 1003
using namespace std;
double dis[maxn][maxn];
int p[maxn];
int n,d;
int x[maxn],y[maxn];
bool vis[maxn];
double get_dis(int i,int j)
{
    double xx=x[i]-x[j];
    double yy=y[i]-y[j];
    return xx*xx+yy*yy;
}
int find(int x)
{
    return p[x]==x?x:(p[x]=find(p[x]));
}
void solve()
{
   memset(vis,0,sizeof(vis));
   for(int i=1;i<=n;i++)p[i]=i;
   char cmd[10];
   int a,b,r;

   while(scanf("%s",cmd)!=EOF)
   {
       if(cmd[0]=='O')
       {
           scanf("%d",&a);
           if(!vis[a])
           {
               r=find(a);
               for(int i=1;i<=n;i++)
               {
                   if(i==a)continue;
                   if(vis[i]&&dis[a][i]<=d&&find(i)!=r)
                   {
                       p[find(i)]=r;
                   }
               }
               vis[a]=1;
           }
           for(int i=1;i<=n;i++)cout<<p[i]<<" ";cout<<endl;
       }
       else
       {
           scanf("%d%d",&a,&b);
           a=find(a);
           b=find(b);
           printf("%s\n",a==b?"SUCCESS":"FAIL");
       }
   }
}

int main()
{
    scanf("%d%d",&n,&d);
    d*=d;
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&x[i],&y[i]);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<i;j++)
        {
            dis[i][j]=dis[j][i]=get_dis(i,j);
        }
    }
    solve();
    return 0;
}