原題:
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;
}