Question Link
题目大意
N台电脑,编号1-N。一个数字d,表示两台计算机的最大通信距离,超过这个距离就无法进行通信。然后分别告诉这些电脑的坐标,电脑已经全部损坏。
接下来有两种操作,第一种O表示这点电脑修好,第二种S,表示测试这两台电脑能不能进行正常的通信。
题目分析
正常并查集拓展写法。
如果一旦修好电脑,那么用这个修好的电脑和其他已经修好的电脑比较,看距离,判断能否通信。s 则是询问操作,判断就行不需要做改变结构任何操作。
AC code
#include<iostream>
#include<queue>
#include<cstring>
#include<string>
#include<sstream>
#include<map>
#include<vector>
#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<stack>
#include<ctime>
using namespace std;
#define rep(i,aa,bb) for(register int i=aa;i<=bb;i++)
#define rrep(i,aa,bb) for(register int i=aa;i>=bb;i--)
#define mset(var,val) memset(var,val,sizeof(var))
#define LL long long
#define eps 0.000001
#define inf 0x7f7f7f7f
#define llinf 1e18
#define exp 0.000001
#define pai 3.141592654
#define random(x) rand()%(x)
#define lowbit(x) x&(-x)
inline int read()
{
int x=0,y=1;char a=getchar();while ( a>'9' || a<'0'){if ( a=='-')y=-1;a=getchar();}
while ( a>='0' && a<='9' ){ x=(x<<3)+(x<<1)+a-'0'; a=getchar();}return x*y;
}
#define N 1009
int x[N],y[N],fa[N];bool is_use[N];
int findfa(int x){
return x == fa[ x ] ? x : findfa(fa[x]);
}
void join(int a,int b){
int afa = findfa(a);
int bfa = findfa(b);
if ( afa != bfa )
fa[ afa ] = bfa;
}
int dist(int a,int b){
return( (x[a]-x[b])*(x[a]-x[b]) + (y[a]-y[b])*(y[a]-y[b]) ) ;
}
int main()
{
// freopen("1.txt","r",stdin);
srand((int)time(0));
// std::ios::sync_with_stdio(false);
int n,d;
n = read(); d = read();
rep(i,1,n){
x[i] = read();
y[i] = read();
}
rep(i,1,n) fa[i] = i ;
mset(is_use,0);
char cmd[8];
while ( scanf("%s",cmd) != EOF ){
if ( cmd[0] == 'O'){
int p = read();
is_use[ p ] = true;
for (int i = 1; i <= n; i++)
if ( is_use[i] && i != p && dist(i,p)<= d*d )
join(i, p);
}
else {
int p,q;
p = read(); q = read();
if ( findfa(p) == findfa(q))
printf("SUCCESS\n");
else
printf("FAIL\n");
}
}
return 0;
}