1. 程式人生 > >poj 1182 食物鏈(經典!種類並查集)

poj 1182 食物鏈(經典!種類並查集)

連結:

原題:

Description

動物王國中有三類動物A,B,C,這三類動物的食物鏈構成了有趣的環形。A吃B, B吃C,C吃A。 
現有N個動物,以1-N編號。每個動物都是A,B,C中的一種,但是我們並不知道它到底是哪一種。 
有人用兩種說法對這N個動物所構成的食物鏈關係進行描述: 
第一種說法是"1 X Y",表示X和Y是同類。 
第二種說法是"2 X Y",表示X吃Y。 
此人對N個動物,用上述兩種說法,一句接一句地說出K句話,這K句話有的是真的,有的是假的。當一句話滿足下列三條之一時,這句話就是假話,否則就是真話。 
1) 當前的話與前面的某些真的話衝突,就是假話; 
2) 當前的話中X或Y比N大,就是假話; 
3) 當前的話表示X吃X,就是假話。 
你的任務是根據給定的N(1 <= N <= 50,000)和K句話(0 <= K <= 100,000),輸出假話的總數。 

Input

第一行是兩個整數N和K,以一個空格分隔。 
以下K行每行是三個正整數 D,X,Y,兩數之間用一個空格隔開,其中D表示說法的種類。 
若D=1,則表示X和Y是同類。 
若D=2,則表示X吃Y。

Output

只有一個整數,表示假話的數目。

Sample Input

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

Sample Output

3
分析與總結: 這題是稍微複雜的種類並查集, 共有3種動物,那麼顯然就應該分為3類。

假設x 吃 y, 那麼x比y的權值大1.因為共有3種動物,那麼權值要分為0,1,2三種。0是根結點, 如果有個節點的權值是1,就表示這個結點是吃根結點的,

如果權值是2,表示2是吃1的(2比1大1),而因為三種動物迴圈吃的關係, 2 也就是被0吃的。

然後在合併時,按照關係分類計算它們的權值。

程式碼:

#include<cstdio>

const int N = 50005;
int n, m, f[N], rank[N];

inline void init(){
    for(int i=0; i<=n; ++i)
        f[i]=i,rank[i]=0;
}

int find(int x){
    if(x==f[x])return x;
    int fa=f[x];
    f[x]=find(f[x]);
    rank[x] = (rank[x]+rank[fa])%3;
    return f[x];
}

inline bool Union(int x,int y,int D){
    int a=find(x), b=find(y);
    if(a==b){
        if(D==1&&rank[x]!=rank[y]) return false;
        if(D==2){
            if(rank[x]==2&&rank[y]!=1)return false;
            if(rank[x]==1&&rank[y]!=0)return false;
            if(rank[x]==0&&rank[y]!=2)return false;
        }
        return true;
    }
    f[a] = b;
    if(D==2){
        rank[a] = (rank[y]-rank[x]+3+1)%3; //注意加上3,因為可能是負數
    }
    else{
        rank[a] = (rank[y]-rank[x]+3)%3;   // 注意加上3,因為可能是負數
    }
    return true;
}

int main(){
    int x,y,d;
    scanf("%d%d",&n,&m);
    init();
    int cnt=0;
    for(int i=0; i<m; ++i){
        scanf("%d%d%d",&d,&x,&y);
        if(x>n || y>n || d==2&&x==y){
            ++cnt; continue;
        }
        if(!Union(x,y,d)) {
            ++cnt;
        }
    }
    printf("%d\n", cnt);
    return 0;
}


——  生命的意義,在於賦予它意義。