poj 1182 食物鏈(並查集)
阿新 • • 發佈:2019-01-23
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
思路:求出每個點到根節點的距離,用距離來判斷出真假。類似hdu 3038題。
#include<iostream> #include<stdio.h> #include<string.h> using namespace std; #define MAXN 50005 int fa[MAXN]; int sum[MAXN];//0吃1,1吃2,2吃0(都是相對根節點來說) int ans; int set_find(int d){ if(fa[d]<0)return d; int tmp=set_find(fa[d]); sum[d]+=sum[fa[d]]; sum[d]=sum[d]%3; return fa[d]=tmp; } void set_join(int D,int x,int y){ int x2=set_find(x); int y2=set_find(y); if(x2!=y2){ fa[y2]=x2; if(D==1){ sum[y2]=sum[x]-sum[y]; sum[y2]=(sum[y2]+3)%3; } else{ sum[y2]=sum[x]+1-sum[y]; sum[y2]=(sum[y2]+3)%3; } } else{ if(D==1){ if(sum[x]!=sum[y])++ans; } else{ if((sum[x]+1)%3!=sum[y])++ans;//X吃X就是假話。當成環的時候滿足sum[x]+1)%3==sum[y] } } } int main(){ int N,K; int D,X,Y; scanf("%d%d",&N,&K); memset(fa,-1,sizeof(fa)); memset(sum,0,sizeof(sum)); ans=0; while(K--){ scanf("%d%d%d",&D,&X,&Y); if(X>N||Y>N){//X或Y比N大,就是假話 ++ans; continue; } set_join(D,X,Y); } printf("%d\n",ans); return 0; }