poj 1182 食物鏈(經典!種類並查集)
阿新 • • 發佈:2019-02-05
連結:
原題:
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
以下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,就表示這個結點是吃根結點的,
然後在合併時,按照關係分類計算它們的權值。
程式碼:
#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; }
—— 生命的意義,在於賦予它意義。