（种类并查集）P2024 [NOI2001] 食物链

题目描述
动物王国中有三类动物 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 句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。

当前的话与前面的某些真的话冲突，就是假话
当前的话中 X 或 Y 比 N 大，就是假话
当前的话表示 X 吃 X，就是假话
你的任务是根据给定的 N 和 K 句话，输出假话的总数。

输入格式
第一行两个整数，N，K，表示有 N 个动物，K 句话。

第二行开始每行一句话（按照题目要求，见样例）

输出格式
一行，一个整数，表示假话的总数。

输入输出样例
输入
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
输出
3
说明/提示
1 ≤ N ≤ 5 ∗ 10^4

1 ≤ K ≤ 10^5

思路
可以将并查集扩大三倍，分为三个集合，分别从1-n表示自己，从n+1-2n表示自己的猎物，从2n+1-3n表示自己的天敌，则就可以根据题意去合并，如果为真话且同类，则将其在三个集合中对应的值合并，如果表示被吃关系，则根据其吃与被吃的关系合并集合。

#include <iostream>
#include<cstdio>
using namespace std;
int pra[150005];
void init()   //并查集常规操作
{
    for(int i=0;i<150005;i++)
        pra[i]=i;
}
int findd(int a)
{
    int root=a;
    while(root!=pra[root])
        root=pra[root];
    while(pra[a]!=root)
    {
        int temp=pra[a];
        pra[a]=root;
        a=temp;
    }
    return root;
}
int un(int a,int b)
{
    a=findd(a);
    b=findd(b);
    if(a!=b)
    {
        pra[a]=b;
        return 1;
    }
    return 0;
}
int main()
{
    int n,k;
    cin>>n>>k;
    int ans=0;
    init();
    for(int i=0;i<k;i++)
    {
        int m,x,y;
        cin>>m>>x>>y;
        if(x>n||y>n)  //如果x或y比n大则为假话
        {
            ans++;
            continue;
        }
        if(m==1)
        {
            if(findd(x)==findd(y+n)||findd(x+n)==findd(y))  //如果y的猎物（y+n）与x在同一集合说明y吃x
            {                                           //如果x的猎物（x+n）与y在同一集合则表示x吃y 均不符合题意
                ans++;
                continue;
            }
            else
            {
                un(x,y);  //如果符合则将x与y在本身，猎物和天敌里全部划分为同类
                un(x+n,y+n);
                un(x+2*n,y+2*n);
            }
        }
        else
        {
            if(x==y)  //由题意若x==y则说明为假话
            {
                ans++;
                continue;
            }
            if(findd(x)==findd(y)||findd(y+n)==findd(x))  //如果x与y为同类或者y吃x则说明是假话
            {
                ans++;
                continue;
            }
            else
            {
                un(x+n,y);  //如果符合则将x的猎物与y合并
                un(y+n+n,x);  //将y的天敌与x合并
                un(x+n+n,y+n);  //则x的天敌为y的猎物
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}