传送门 Wireless Network
题意 : 给你n个坏机器的坐标,和最大沟通距离 d;给定两种操作
‘S’ 和 'O' 第一种操作询问两个机器是否联系,第二种修复这个编号的机器
联系具有传递性 ,那么就用并查集 ; so :
#include <stdio.h>
#include <math.h>
#include <vector>
using namespace std ;
const int maxn = 1e3+5 ;
struct node
{
int x , y ,flag ;
} pos[maxn] ;
int ufs[maxn] , rnk[maxn] ;
int n , d ;
vector <int> num ; // 记录修好的机器坐标
void init( ) // 初始化
{
for( int i = 1 ; i <= n ; i++) rnk[i] = 1 , ufs[i] = i , pos[i].flag = 0 ;
}
void input ( ) // 初始化
{
for ( int i = 1 ; i <= n ; i++ ) scanf( "%d %d" , &pos[i].x ,&pos[i].y ) ;
}
int find_set( int x ) // 查找
{
if( x != ufs[x] )
ufs[x] = find_set(ufs[x] ) ;
return ufs[x] ;
}
void union_set( int a , int b ) // 合并
{
a = find_set(a) ;
b = find_set(b) ;
if( a == b ) return ;
if( rnk[a] > rnk[b] ) ufs[b] = a ; // 按秩合并
else
{
if( rnk[a] == rnk[b] ) rnk[a] ++ ;
ufs[a] = b ;
}
}
bool check ( node a , node b ) // 检查是否可以联系
{
double dis = sqrt ( pow( (double)(a.x-b.x) , 2 ) + pow( (double)(a.y-b.y) , 2 ) ) ;
if( dis <= d ) return true ;
return false ;
}
int main( )
{
char ch[10] ;
int a , b ;
scanf( "%d %d" , &n , &d) ;
init ( ) ;
input( ) ;
while ( scanf( "%s" , ch) != EOF )
{
if( ch[0] == 'O' )
{
scanf( "%d" , &a) ;
pos[a].flag = 1 ;
for( int i = 0 ; i < num.size( ) ; i++ )
if ( check( pos[num[i]] , pos[a] ) )
union_set( num[i] , a ) ;
num.push_back(a) ;
}
else
{
scanf( "%d %d" , &a , &b ) ;
if( !pos[a].flag || !pos[b].flag ) // 未修复那么就无法联系
printf( "FAIL\n") ;
else if ( find_set(a) != find_set(b) ) // 不在同一个子集就无法联系
printf( "FAIL\n") ;
else
printf( "SUCCESS\n") ;
}
}
return 0 ;
}