题意:
题目大意是说有一些电脑,编号为1到N,现在这些电脑坏了,无法相互连通,我们需要维修,输入首先输入N和d,N表示有多少台电脑,d表示两台已维修好的电脑若它们之间的距离小于等于d,则两台电脑可以互通。接下来输入N行,每行输入a,b两个数,N行中的第i行表示编号为i的电脑的坐标(用来求两台电脑的距离),在接下来的输入各种操作,O a表示编号为a的电脑被维修好了,S a b则表示询问编号为a和b的电脑能不能互通,若能则输出SUCCESS,若不能则输出FAIL。
距离:s=sqrt(b*b+c*c)
因为有10s,所以并不难
关键部分:
if(a!=i && check(i,a) && vis[i]){
unio(i,a);
}
AC Code:
#include<iostream>
#include<cstdio>
#include<queue>
#include<math.h>
using namespace std;
int n;
int d;
int x[1010],y[1010],vis[1010],a,u,v;
int fa[1010];
char op;
int findfa(int x){
if(fa[x]==x) return x;
return fa[x]=findfa(fa[x]);
}
bool check(int l,int r){
if(1.0*sqrt(1.0*(x[l]-x[r])*(x[l]-x[r]) + 1.0*(y[l]-y[r])*(y[l]-y[r]))<=1.0*d){
// Tip: 不是d*d,因为已经sqrt过了
return true;
}
return false;
}
void unio(int x,int y){
int Lfa=findfa(x);
int Rfa=findfa(y);
if(Lfa!=Rfa){
fa[Rfa]=Lfa;
}
}
int main(){
scanf("%d%d",&n,&d);
for(int i=1;i<=n;i++){
scanf("%d%d",&x[i],&y[i]);
}
for(int i=1;i<=n;i++){
fa[i]=i;
vis[i]=0;
}
while(~scanf("%c",&op)){
if(op=='O'){
scanf("%d",&a);
vis[a]=1;
for(int i=1;i<=n;i++){
if(a!=i && check(i,a) && vis[i]){
unio(i,a);
}
}
}
if(op=='S'){
scanf("%d%d",&u,&v);
if(findfa(u)==findfa(v)){
printf("SUCCESS\n");
} else {
printf("FAIL\n");
}
}
}
}