poj 2912 Rochambeau (并查集+枚举)

http://poj.org/problem?id=2912

枚举＋关系并查集，并查集与1182相似，枚举每个小孩为judge时的情况，若当前枚举情况下每个round都是正确的，则当前枚举编号可能是judge。若只找到一个judge的可能，则找出排除其他编号所需最多的round数，两者即为输出。

做了一整天，WA了9次，原因是偏移量r公式搞错一个地方，列了好多次表才搞定。

code:

#include<cstdio>
#include<cstring>
using  namespace std ;
int f[ 501] ;
int r[ 501] ;
int x[ 2001] ;
int y[ 2001] ;
int ans[ 501] ;
void make_set( int n){
     for( int i= 0; i<n; i++){
        f[i] = i ;
        r[i] =  0 ;
    }
}
int find_set( int x){
     int temp ;
     if(x==f[x])
         return x ;
    temp = f[x] ;
    f[x] = find_set(temp) ;
    r[x] = (r[temp]+r[x]) %  3 ;
     return f[x] ;
}
void union_set( int x,  int y,  int fx,  int fy,  int d){
    f[fy] = fx ;
    r[fy] = ( 3+d+r[x]-r[y]+r[fx]) %  3 ;
}
int main(){
     int n, m, i, j, fx, fy, p, s, count ;
     int d[ 2001] ;
     while(~scanf( " %d%d ", &n, &m)){
        memset(ans,  0,  sizeof(ans)) ;
         for(i= 0; i<m; i++){
            scanf( " %d ", &x[i]) ;
             char c = getchar() ;
            scanf( " %d ", &y[i]) ;
             if(c== ' = ')  d[i] =  0 ;
             else  if(c== ' < ') d[i] =  1 ;
             else d[i] =  2 ;
        }
         for(i= 0; i<n; i++){
            make_set(n) ;
             for(j= 0; j<m; j++){
                 if(x[j]==i||y[j]==i)     continue ;
                fx = find_set(x[j]) ;
                fy = find_set(y[j]) ;
                 if(fx!=fy)
                    union_set(x[j], y[j], fx, fy, d[j]) ;
                 else
                     if((r[x[j]]+d[j])% 3!=r[y[j]]){
                        ans[i] = j +  1 ;         // 记录出错位置
                         break ;
                    }
            }
        }
        s =  0, count =  0 ;
         for(i= 0; i<n; i++){
             if(ans[i]== 0){
                count ++ ;
                p = i ;
            }
             if(ans[i]>s)
                s = ans[i] ;
        }
         if(count== 0)    printf( " Impossible\n ") ;
         else  if(count> 1)    printf( " Can not determine\n ") ;
         else    printf( " Player %d can be determined to be the judge after %d lines\n ", p, s) ;
    }
     return  0 ;

}