题意:距离d内的两台电脑互联。现在通过修理和查看操作,求查看时两台电脑是否互联?
分析:这是并查集的基本应用,两台修好的电脑若距离d内则加入合并。
#include<cstdio>
#include<algorithm>
#include<cstring>
#define MAX_N 1005
using namespace std;
struct node
{
int x,y;
}computer[MAX_N];
int par[MAX_N];
int r[MAX_N];
int d,n;
bool repaired[MAX_N];
bool connected[MAX_N][MAX_N];
void init(int n)
{
for(int i=0;i<=n;i++)
{
par[i]=i;
r[i]=0;
for(int j=i;j<=n;j++)
{
if((computer[i].x-computer[j].x)*(computer[i].x-computer[j].x)+(computer[i].y-computer[j].y)*(computer[i].y-computer[j].y)<=d*d)
connected[i][j]=connected[j][i]=true;
else
connected[i][j]=connected[j][i]=false;
}
}
return ;
}
int find(int x)
{
if(x==par[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(r[x]<r[y])
par[x]=y;
else
{
par[y]=x;
if(r[x]==r[y])
r[x]++;
}
return ;
}
bool same(int x,int y)
{
return find(x)==find(y);
}
int main(void)
{
char op;
while(scanf("%d%d",&n,&d)==2)
{
memset(repaired,false,sizeof(repaired));
for(int i=0;i<n;i++)
scanf("%d%d",&computer[i].x,&computer[i].y);
init(n);
int p,q;
while(scanf("%s",&op)==1)
{
if(op=='O')
{
scanf("%d",&p);
p--;
repaired[p]=true;
for(int i=0;i<n;i++)
{
if(i==p)
continue;
if(repaired[i]&&connected[i][p])
unite(p,i);
}
}
else
{
scanf("%d%d",&p,&q);
p--;
q--;
if(same(p,q))
printf("SUCCESS\n");
else
printf("FAIL\n");
}
}
}
return 0;
}