【并查集】noi2001食物链

P2024 [NOI2001]食物链

//这是一道比我年纪大的题啊啊啊啊啊QAQ

加权并查集  三倍并查集好厉害qwq

图源洛谷题解

贴代码qwq

#include<cstdio>
#include<iostream>
using namespace std;
int n, k, opt, x, y, ans = 0, f[150010];
void init() { for(int i = 1; i <= n*3; i++) f[i] = i;}
//开三倍并查集，一倍存自己的同类， 二倍存自己的猎物， 三倍存自己的天敌 
int find(int x) {
    if(x != f[x]) f[x] = find(f[x]);
    return f[x];
} 
int merge(int u, int v) {
    int r1 = find(u), r2 = find(v);
    f[r1] = r2;
}
int main() {
    scanf("%d%d",&n,&k);
    init();
    while(k--) {
        scanf("%d%d%d",&opt,&x,&y);
        if(x>n||y>n) ans++;
        else if(opt == 1) {
            if(find(x+n)==find(y)||find(x+2*n)==find(y)) ans++;
            //如果x的猎物是y 或者 如果x的天敌是y 
            else {
                merge(x, y), merge(x+n, y+n), merge(x+2*n, y+2*n);
            }
        }
        else {
            if(x == y) ans++;
            else {
                if(find(x)==find(y)||find(x+2*n)==find(y)) {ans++; continue;}
                //如果x和y是同类 如果x的天敌是y  
                  merge(x+n,y), merge(x,y+2*n), merge(x+2*n,y+n);
                  //如果是真话， 则x的猎物是y， x和y的天敌是同类， x的天敌和y的猎物是同类  
            }
        }
    }    
    printf("%d",ans);
    return 0;
}

加油加油ヾ(◍°∇°◍)ﾉﾞ