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

思路:求出每個點到根節點的距離,用距離來判斷出真假。類似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;
}