食物链 (种类并查集)

食物链

题目描述
动物王国中有三类动物 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 1 3表示1 和 3是同类,那么merge(1,3)来维护本身关系,merge(1 + 3,3 + 3)维护被捕食者的关系,merge(1 + 6,3 + 6)维护天敌的关系
2 1 2表示 1 和 2是捕食这的关系,那么(1 + 3, 2)维护捕食者的关系,merge(1,2 + 6)维护猎物的关系,merge(1 + 6,2 + 3)维护天敌的关系

AC代码

#include<bits/stdc++.h>
const int maxn = 2e5 + 5;
int n,k,i,z,x,y,ans;
int pre[maxn];
int find(int x){//找祖宗函数
    if(x==pre[x]) return x;
    return pre[x] = find(pre[x]);
}
void merge(int x,int y){//合并函数
    int fx = find(x);
    int fy = find(y);
    if(fx!=fy)
        pre[fx]=fy;
}
int main(){
    scanf("%d %d",&n,&k);
    for(i=1;i<=3*n;i++){
        pre[i]=i;
    }
    for(i=1;i<=k;i++){
        scanf("%d %d %d",&z,&x,&y);
        if(x>n||y>n){//数据超出范围显然直接弹出
            ans++;
            continue;
        }
        if(z==1){
            if(find(x+n)==find(y)||find(x+2*n)==find(y)){//如果x和y是捕食关系显然也不行
                ans++;
            }
            else{
                merge(x,y);//x和y是一个种类
                merge(x+n,y+n);//x和y的猎物也是一个种类
                merge(x+2*n,y+2*n);//x和y的天敌也是一个种类
            }
        }
        else if(z==2){
            if(find(x)==find(y)||find(x+2*n)==find(y)){//如果x和y是一个种类或者y是x的天敌显然不行
                ans++;
            }
            else{
                merge(x+n,y);//x的猎物是y
                merge(x+2*n,y+n);//x的天敌是y的猎物
                merge(x,y+2*n);//x是y的天敌
            }
        }
    }
    printf("%d",ans);
    return 0;
}