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


思路:很明顯的種類並查集。需要分三種。用虛擬節點做,開三倍pre[]。別的和兩種的大同小異。

坑點:如果沒理解透虛擬節點是如何執行的,在判斷 “當前的話與前面的某些真的話衝突,就是假話” 時可能會直接寫if(!same(X,Y))


#include<cstdio>
using namespace std;
int N,K;
int pre[150005];

int find(int x){
    while(pre[x]!=x){
        int r=pre[x];
        pre[x]=pre[r];
        x=r;        
    }
    return x;
}

void merge(int x,int y){
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy) pre[fx]=fy;
}

int
same(int x,int y){ return find(x)==find(y); } int main(){ int D,X,Y; scanf("%d%d",&N,&K); for(int i=1;i<=3*N;i++) pre[i]=i; int ans=0; while(K--){ scanf("%d%d%d",&D,&X,&Y); if(X>N||Y>N){ ans++; continue; } if(D==1){ if(same(X,Y+N)||same(X,Y+2*N)){ ans++; continue; } merge(X,Y); merge(X+N,Y+N); merge(X+2*N,Y+2*N); } else{ if(same(X,Y)||same(X,Y+N)){ ans++; continue; } merge(X,Y+N*2); merge(X+N,Y); merge(X+2*N,Y+N); } } printf("%d",ans); return 0; }