程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> poj 1182 食物鏈(經典!種類並查集)

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

編輯:C++入門知識

原題:
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吃的。
然後在合並時,按照關系分類計算它們的權值。


代碼:
[cpp] 
#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; 


 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved