NKOJ1209并查集【NOI2001 Day1 T3】食物链

问题描述

动物王国中有三类动物 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），输出假话的总数。

输入格式

第一行是两个整数N和K，以一个空格分隔。
以下K行每行是三个正整数D，X，Y，两数之间用一个空格隔开，其中 D 表示说法的种类。
若D=1，则表示X和Y是同类。
若D=2，则表示X吃Y。

输出格式

只有一个整数，表示假话的数目。

样例输入

输入文件   
100 7
1 101 1    
2 1 2       
2 2 3       
2 3 3       
1 1 3       
2 3 1       
1 5 5       

样例输出

3

提示

对7句话的分析
100 7
1 101 1 假话
2 1 2 真话
2 2 3 真话
2 3 3 假话
1 1 3 假话
2 3 1 真话
1 5 5 真话

思路分析：

rala[x]表示x与父节点的关系：

0表示同类，1表示被父节点吃，2表示吃父节点。

那么 输入的关系 d  就应该对应  (d-1)的情况；

那么父亲节点与儿子节点x的关系就是 （3-rala[x]），这一点在合并集合的时候会用到。 

注意：这里的0、1、2不是乱取的，不能换成1，2，3。这和题目假设有很大关系，例如：

与父亲同类+父亲与爷爷同类=同类 ==> 0+0=0;

同类+父节点被爷爷节点吃=被爷爷节点吃 ==> 0+1=1；

被父亲节点吃+父亲节点吃爷爷节点 =与爷爷同类 ==> (1+2)%3==0;

其他的情况以此类推，于是有：

与爷爷的关系=（与父亲的关系+父亲与爷爷的关系）%3；

依次迭代下去即可；

正确代码：

#include <bits/stdc++.h>
using namespace std;
const int NN=5e4+10;
int n,m;
int a[3*NN];
int find(int x){ 
    if(x!=a[x])
        a[x]=find(a[x]);
    return a[x];
}
void join(int x,int y){ 
    int fx=find(x),fy=find(y); 
    if(fx!=fy){
        a[fx]=fy;
    }
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=3*n;i++) a[i]=i; 
    int t,x,y;
    int coun=0;
    while(m--){
        scanf("%d%d%d",&t,&x,&y);
        if(x>n||y>n){
            coun++;
            continue;
        }
        if(t==1){
            if(find(x+n)==find(y)||find(x+2*n)==find(y)) coun++;
			else{
                join(x,y);join(x+n,y+n);join(x+2*n,y+2*n); 
            }
        }
		else{
            if(x==y) coun++;
			else if(find(x)==find(y)||find(x+2*n)== find(y)){ 
                coun++;
            }
			else{
                join(x+n,y);join(x+2*n,y+n);join(x,y+2*n); 
            }
        }
    }
    cout<<coun;
    return 0;
}