luogu2024 食物链 (并查集)

把一个点拆成三个，分别对应它的同类、它的猎物和它的天敌，这样的话（以下的相等都是并查集意义上的）：

如果令a,b同类，那么a的猎物不能是b的同类，a的天敌不能是b的同类

如果令a吃b，那么a的同类不能是b的同类、a的天敌不能是b的同类。

要令a,b同类，就使（a和b）的同类、猎物、天敌分别相等

要令a吃b，就使a的猎物是b的同类，b的天敌是a的同类，a的天敌是b的猎物（题目中要求A吃B，B吃C，C吃A）

#include<bits/stdc++.h>
#define pa pair<int,int>
#define ll long long
using namespace std;
const int maxn=50050;

inline ll rd(){
    ll x=0;char c=getchar();int neg=1;
    while(c<'0'||c>'9'){if(c=='-') neg=-1;c=getchar();}
    while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*neg;
}

int N,K,fa[maxn*3];

int getfa(int x){return x==fa[x]?x:fa[x]=getfa(fa[x]);}
void add(int a,int b){
    fa[getfa(a)]=getfa(b);
}

int main(){
    //freopen("testdata.in","r",stdin);
    int i,j,k,ans=0;
    N=rd();K=rd();
    for(i=1;i<=N*3;i++) fa[i]=i;
    for(i=1;i<=K;i++){
        int a=rd(),b=rd(),c=rd();
        if(b>N||c>N) {++ans;continue;}
        if(a==1){
            if(getfa(b)==getfa(c+N)||getfa(b)==getfa(c+2*N)){++ans;continue;}
            add(b,c);add(b+N,c+N);add(b+2*N,c+2*N);
        }else{
            if(getfa(b)==getfa(c)||getfa(b+2*N)==getfa(c)){++ans;continue;}
            add(b+N,c);add(b,c+2*N);add(b+2*N,c+N);
        }
    }
    printf("%d\n",ans);
    return 0;
}