poj1182解题报告

题意：中文题，自己读吧

题解：并查集，只不过增加了一个当前点到与根节点的关系

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std ;

const int MAXN = 50005 ;
int   N ;

int   father[MAXN] ;
int   rank[MAXN*2] ;

int  change( int X )
{
    switch( X )
    {
        case 0  : return 0  ;
        case 1  : return 1  ;
        case 2  : return -1 ;
        case -1 : return -1 ;
        case -2 : return 1  ;
    }
}

int  find( int X )
{
    if( father[X] == X ) return X ;
    int  t = father[X] ;
    father[X] = find( father[X] ) ;
    rank[X] = change( rank[X] + rank[t] ) ;
    return father[X] ;
}

bool  function( int D , int X , int Y )
{
    if( X > N || Y > N ) return true ;

    int  f1 = find( X ) ;
    int  f2 = find( Y ) ;

    if( f1 != f2 )
    {
        father[f1] = f2 ;
        if( D == 1 )
        {
            if( rank[X] == rank[Y] ) { rank[f1] = 0 ; return 0 ; }
            if( rank[X] == 0 ) { rank[f1] = rank[Y] ; return 0 ; }
            if( rank[X] == 1 && rank[Y] == 0 ) { rank[f1] = -1  ; return 0 ; }
            if( rank[X] == 1 && rank[Y] == -1 ) { rank[f1] = 1  ; return 0 ; }
            if( rank[X] == -1 && rank[Y] == 0 ) { rank[f1] = 1  ; return 0 ; }
            if( rank[X] == -1 && rank[Y] == 1 ) { rank[f1] = -1 ; return 0 ; }
        }
        else
        {
            if( rank[X] == 0 && rank[Y] == 0  ) { rank[f1] = 1  ; return 0 ; }
            if( rank[X] == 0 && rank[Y] == 1  ) { rank[f1] = -1 ; return 0 ; }
            if( rank[X] == 0 && rank[Y] == -1 ) { rank[f1] = 0  ; return 0 ; }
            if( rank[X] == 1 && rank[Y] == 0  ) { rank[f1] = 0  ; return 0 ; }
            if( rank[X] == 1 && rank[Y] == 1  ) { rank[f1] = 1  ; return 0 ; }
            if( rank[X] == 1 && rank[Y] == -1 ) { rank[f1] = -1 ; return 0 ; }
            if( rank[X] == -1 && rank[Y] == 0 ) { rank[f1] = -1 ; return 0 ; }
            if( rank[X] == -1 && rank[Y] == 1 ) { rank[f1] = 0  ; return 0 ; }
            if( rank[X] == -1 && rank[Y] == -1 ) { rank[f1] = 1 ; return 0 ; }
        }
    }
    else
    {
        if( D == 1 )
        {
            if( rank[X] != rank[Y] ) return 1 ;
        }
        else
        {
            if( rank[X] == rank[Y] ) return 1 ;
            if( rank[X] == -1 && rank[Y] == 0 ) return 1 ;
            if( rank[X] == 0 && rank[Y] == 1  ) return 1 ;
            if( rank[X] == 1 && rank[Y] == -1 ) return 1 ;
        }
        return 0 ;
    }
}

int  main()
{
    int  K , ans = 0 ;
    int  D , X , Y   ;

    scanf( "%d%d" , & N , & K ) ;

    memset( rank , 0 , sizeof( rank ) ) ;
    for( int i = 0 ; i <= N ; i ++ ) father[i] = i ;

    while( K -- )
    {
        scanf( "%d%d%d" , & D , & X , & Y ) ;
        ans += function( D , X , Y ) ;
    }

    printf( "%d\n" , ans ) ;

    return 0 ;
}