poj Wireless Network

 传送门  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 ;
}