一些电脑,编号为1到N,现在这些电脑坏了,无法相互连通,维修,
输入首先输入N和d,N电脑台数,
d两台已维修好的电脑
若它们之间的距离小于等于d,则两台电脑可以互通。
接下来输入N行,每行输入a,b两个数,
N行中的第i行表示编号为i的电脑的坐标(用来求两台电脑的距离)
O a表示编号为a的电脑被维修好了,
S a b则表示询问编号为a和b的电脑能不能互通,若能则输出SUCCESS,若不能则输出FAIL。
4 1
0 1
0 2
0 3
0 4
O 1
O 2
O 4
S 1 4
O 3
S 1 4
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
int fa[1005],vis[1005];//标记此电脑是否被修理
struct num{
int x,y;
}num[1005];
int find(int x)
{
if(fa[x]==x)
return x;
return fa[x]=find(fa[x]);
}
bool same(int x,int y){
return find(x)==find(y) ;
}
void bind(int x,int y)
{
int fx=find(x);
int fy=find(y);
if(fx!=fy)
fa[fx]=fy;
}
double dis(int num1,int num2){
return sqrt(1.0*(num[num2].x-num[num1].x)*(num[num2].x-num[num1].x)+1.0*(num[num2].y-num[num1].y)*(num[num2].y-num[num1].y));
}
int main()
{
int num1,num2,numx,n,d;
char c;
scanf("%d %d",&n,&d);
for(int i=1;i<=n;i++){
fa[i]=i;
vis[i]=0;
}
for(int i=1;i<=n;i++){
scanf("%d %d",&num[i].x,&num[i].y);
}
while(scanf("%c",&c)!=EOF)
{
if(c=='O'){
scanf("%d",&numx);
vis[numx]=1;
for(int i=1;i<=n;i++){
if(i!=numx&&vis[i]&&dis(i,numx)<=d)
bind(i,numx);
}
}
else if(c=='S'){
scanf("%d %d",&num1,&num2);
if(same(num1,num2))
printf("SUCCESS\n");
else printf(("FAIL\n"));
}
}
return 0;
}