poj 1182 食物链 (并查集)

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

重点依旧是用rank记录相对于根节点的关系，并及时跟新rank的值。

code:

#include<cstdio>
using namespace std ;
int f[50001] ;
int r[50001] ;//与根节点的关系
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 ;
    /*
        r[x]是相对于x原根节点temp的关系值，现在相对根节点已不是temp
        故更新r[x]值时要加上temp相对于其根节点的差值r[temp]
        递归调用，查找一次x可将在其前入集的点全部更新
    */
    return f[x] ;
}
void Union(int x, int y, int fx, int fy, int d){
    f[fy] = fx ;
    r[fy] = (d+2+r[x]-r[y]) % 3 ;
    /*
        r[]为此元素与根节点关系，1为被吃，2为吃
        find_set()后，如果x已经入集，则fx为根节点，若未入，则fx为其本身
        无论哪种情况r[fx],r[fy]均为0
        将fx作为根节点，则r[x]不变，因为r[fx]恒为0
        若x吃y，则r[y] = r[x]+1
        若为同类，则r[y] = r[x]+0
        列表可得出r[fy]的改变量
    */
}
int main(){
    int n, k, i, d, x, y, fail, fx, fy ;
    scanf("%d%d", &n, &k) ;
    fail = 0 ;
    for(i=0; i<=n; i++){
        f[i] = i ;
        r[i] = 0 ;
    }
    for(i=0; i<k; i++){
        scanf("%d%d%d", &d, &x, &y) ;
        if(x>n||y>n||(d==2&&x==y)){
            fail ++ ;
            //printf("%d %d\n", x, y) ;
            continue ;
        }
        fx = find_Set(x) ;
        fy = find_Set(y) ;
        if(fx!=fy)
            Union(x, y, fx, fy, d) ;
        else{
            if(d==1&&r[x]!=r[y]){//与root关系不同
                fail ++ ;
                //printf("%d %d\n", x, y) ;
            }
            if(d==2&&(r[x]-r[y]+3)%3!=2){//列举x吃y情况下r所有可能，得出结论
                fail ++ ;
                //printf("%d %d\n", x, y) ;
            }
        }
    }
    printf("%d\n", fail) ;
    return 0 ;

}