洛谷P2024 食物链

好，经典扩展域并查集(冰茶几)。

因为有三个动物形成环(这不生物)，所以并查集开3倍。

x - y 表示同类，x - (y + n) 表示 y 吃 x

然后瞎搞搞即可。

思考：如果不是环，就是个A <- B <- C的食物链呢？如果还有A吃C呢？

据说是bitset

#include <cstdio>

const int N = 50010;

int fa[N * 3], n;

int find(int x) {
    if(x == fa[x]) {
        return x;
    }
    return fa[x] = find(fa[x]);
}

inline void merge(int x, int y) {
    fa[find(x)] = find(y);
    return;
}

inline bool check(int f, int x, int y) {
    if(f == 1) {
        return find(x) != find(y + n) && find(x) != find(y + n + n);
    }
    return find(x) != find(y) && find(y + n) != find(x);
}

inline void add(int f, int x, int y) {
    if(f == 1) {
        if(find(x) == find(y)) {
            return;
        }
        merge(x, y);
        merge(x + n, y + n);
        merge(x + n + n, y + n + n);
    }
    else {
        if(find(x + n) == find(y)) {
            return;
        }
        merge(x + n, y);
        merge(x + n + n, y + n);
        merge(x, y + n + n);
    }
    return;
}

int main() {
    int m;
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n * 3; i++) {
        fa[i] = i;
    }
    int f, x, y, ans = 0;
    for(int i = 1; i <= m; i++) {
        scanf("%d%d%d", &f, &x, &y);
        if(x > n || y > n) {
            ans++;
            continue;
        }
        else if(f == 2 && x == y) {
            ans++;
            continue;
        }
        if(check(f, x, y)) {
            add(f, x, y);
        }
        else {
            ans++;
        }
    }
    printf("%d", ans);
    return 0;
}