洛谷2024 食物链 并查集

问题描述

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

并查集基本递归操作 洛谷里题解写的很清楚，不多赘述

#include<bits/stdc++.h>
using namespace std;
#define maxn 100001
int v[maxn*3];
int n,m,a,b,c;
int ans;
int find(int x){
    if(v[x]!=x)return find(v[x]);
    else return x;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=3*n;i++)v[i]=i;
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&a,&b,&c);
        if(b>n||c>n){ans++;continue;}
        if(a==1){
            if(find(b+n)==find(c)||find(c+n)==find(b))ans++;
            else{
                v[find(b)]=v[find(c)];v[find(b+n)]=v[find(c+n)];v[find(b+2*n)]=v[find(c+2*n)];
            }
        }
        else
            {
            if(find(b)==find(c)||find(c+n)==find(b))ans++;
            else
                {
                v[find(b+n)]=v[find(c)];v[find(c+2*n)]=v[find(b)];v[find(c+n)]=v[find(b+2*n)];
                }    
            }
    }
    printf("%d",ans);
    return 0;
}