[kuangbin带你飞]专题五 并查集 A POJ 2236

题目地址：https://vjudge.net/contest/66964#problem/A

思路：并查集模板题。

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=1001+10;

struct point{
int x,y;
};
point a[maxn];
int fa[maxn];
int n,d;
bool vis[maxn];

int get(int s,int e)
{
return (a[s].x-a[e].x)*(a[s].x-a[e].x)+(a[s].y-a[e].y)*(a[s].y-a[e].y);
}

int find(int p)
{
int t=p;
if(p!=fa[p])
{
fa[p]=find(fa[p]);
}
return fa[p];
}

void merge(int p,int q)
{
p=find(p);
q=find(q);
if(p!=q)
{
fa[q]=p;
}
}

int main()
{
scanf("%d%d",&n,&d);
memset(vis,false,sizeof(vis));
for(int i=1;i<=n;i++)
{
scanf("%d%d",&a[i].x,&a[i].y);
}
char s[2];
for(int i=0;i<=n;i++)
fa[i]=i;
while(~scanf("%s",s))
{
if(s[0]=='O')
{
int temp;
scanf("%d",&temp);
vis[temp]=true;
for(int i=1;i<=n;i++)
{
int dis=get(temp,i);
if(dis<=d*d && vis[i])
{
merge(i,temp);
}
}
}
else
{
int p,q;
scanf("%d%d",&p,&q);
p=find(p);
q=find(q);
if(p!=q)
printf("FAIL\n");
else
printf("SUCCESS\n");
}
}

}