题目
考察并查集。最开始写这道题时,想着进行距离不超过d的点放在一组,最后通过数组vis对’O’输入的数字进行记录。但是尝试很久,发现不行。因为距离大于d的两个点之间可进行多次转移,而转移的方式很多,这是就无法判断了。所以改用边加入点,边合并点。
并查集是一种用来管理元素分组情况的数据结构。并查集可以高效地进行如下操作。不过需要注意并查集虽然可以进行合并操作,但却无法进行分割操作。
查询元素a和元素b是否属于同一组。
合并元素a和元素b所在的组。
#include <cstdio>
#include <cmath>
#include <vector>
using namespace std;
struct node{
int x, y, index;
};
vector<node> v;
vector<int> v1;
const int maxn=1010;
int father[maxn];
void init(int n){
for(int i=1; i<=n; i++){
father[i]=i;
}
}
int find(int r){
while(r!=father[r]){
r=father[r];
}
return r;
}
int main()
{
int n, d;
scanf("%d%d", &n, &d);
init(n);
for(int i=1; i<=n; i++){
node temp;
scanf("%d%d", &temp.x, &temp.y);
temp.index=i;
v.push_back(temp);
}
getchar();
char judge;
while(scanf("%c", &judge)!=EOF){
int a, c;
if(judge=='O'){
scanf("%d", &a);
if(v1.size()==0){
v1.push_back(a);
} else {
int b;
for(int i=0; i<v1.size(); i++){
b=(int)pow(1.0*abs(v[v1[i]-1].x-v[a-1].x), 2.0)+(int)pow(1.0*abs(v[v1[i]-1].y-v[a-1].y), 2.0);
if(b<=(int)pow(1.0*d, 2.0)){
int fx=find(v1[i]);
int fy=find(a);
if(fx!=fy){
father[fx]=fy;
}
}
}
v1.push_back(a);
}
} else {
scanf("%d%d", &a, &c);
int fx=find(a);
int fy=find(c);
if(fx!=fy){
printf("FAIL\n");
} else {
printf("SUCCESS\n");
}
}
getchar();
}
return 0;
}