POJ 1182 食物链 并查集变式

动物王国中有三类动物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

思路：

通过建立一个并查集来实现这三种关系，建一个re数组，0表示与根节点同类，1表示被根节点捕食，2表示捕食根节点。

然后边建立并查集边进行判定。

判定就需要对根节点是否相同进行判定了。

即查找根节点操作。查找的操作不仅仅需要进行路径压缩，而且需要改变re[x]的值。这里有个等式，即re[x]=(re[x]+re[temp])%3.

temp为x的父节点.这个关系可以通过画一个很简单的草图推出。跟向量运算差不多。

如果两个元素的根节点相同，那么就需要对他们进行判定了。

1. 如果输入的是1，即两个元素同类，那么只要看看当前两个集合是否与根节点的关系相同就可以。
2. 如果输入的是2，即X吃Y，那么这中情况下的反例就是X与Y同类，或Y吃X。同类的话，他们与根节点的关系一定相同。即re[x]==re[y]，Y吃X这个推起来有点复杂，这里给出两种情况。（1）X与根节点同类，且Y吃根节点即,re[x]=0,re[y]=2;（2）X吃根节点，根节点吃Y，则Y吃根节点。即 re[x]=2,re[y]=1.通过上面的两种情况发现,re[x]=(re[y]+1)%3.

如果两个子节点的根节点不相同，那么就需要联合操作了。联合的过程其实也是向量的运算。 即re[temp2]=(re[x]+d-re[y]+3)%3;

注意这里的d与不能直接用题目中的D进行替换， 因为这里的1，2与题目中的1，,2的含义不同， 不能直接进行运算， 只需要在原基础上减1即可。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=50005;
int n,k;
int re[maxn];
int a[maxn];
int finds(int x)
{
    int temp=a[x];
    if(a[x]==x)
        return x;
    int par=finds(a[x]);
    re[x]=(re[x]+re[temp])%3;
    a[x]=par;
    return par;
}
int main()
{
    scanf("%d%d",&n,&k);
    for (int i=1;i<=n;i++)
    {
        re[i]=0;
        a[i]=i;
    }
    int num=0;
    for (int i=0;i<k;i++)
    {
        int t,x,y;
        scanf("%d%d%d",&t,&x,&y);
        int temp1=finds(x);
        int temp2=finds(y);
        if(x>n||y>n||(t==2&&x==y))
             num++;
        else if(temp1==temp2)
        {
           if(t==1&&re[x]!=re[y])
                  num++;
           if(t==2&&(re[x]==re[y]||re[x]==(re[y]+1)%3))
                 num++;
        }
        else if(temp1!=temp2)
        {
            re[temp2]=(t-1+re[x]-re[y]+3)%3;
            a[temp2]=temp1;
        }
    }
    printf("%d\n",num);
    return 0;
}