POJ1182-食物链

该题有三种动物，我们可以把一个并查集分成三份：x，x+n， x+2n分别表示A，B，C三种动物。

当x与y是同类时，x与y可能是三种动物中的任意一种，则将(x, y)、(x+n, y+n)、(x+2n, y+2n)合并。

当x吃y时，x与y可能是三种捕食关系中的一种，x，y为异类，则合并A吃B(x, y+n)、B吃C(x+n, y+2n)、C吃A(x+2n, y)。

于是，当x，y处于同一个n内表示x，y为同类。

x，y跨过不同的n即表示x吃y。

对逐个语句进行分析时，先判断语句类型。

当语句表示x， y同类时，检查x，y是否为捕食关系。

当语句表示x吃y时，检查x，y是否是同类和x是否能够吃y。

#include <cstdio>

const int MAX_N = 5*10e3;
const int MAX_K = 10e5;

int par[3*MAX_N+5];
int rank[3*MAX_K+5];
int T[MAX_K];
int X[MAX_K];
int Y[MAX_K];

void init(int n)
{
    for (int i = 0; i < n; i++) {
        par[i] = i;
        rank[i] = 0;
    }
}

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

void unite(int x, int y)
{
    x = find(x);
    y = find(y);
    if (x == y) {
        return;
    }
    
    if (rank[x] < rank[y]) {
        par[x] = y;
    }
    else {
        par[y] = x;
        if (rank[x] == rank[y]) {
            rank[x]++;
        }
    }
}

bool same(int x, int y)
{
    return find(x) == find(y);
}

int main()
{
    int n, k;
    scanf("%d%d", &n, &k);
    
    for (int i = 0; i < k; i++) {
        scanf("%d%d%d", &T[i], &X[i], &Y[i]);
    }
    
    init(3 * n);
    
    int ans = 0;
    for (int i = 0; i < k; i++) {
        int t = T[i];
        int x = X[i] - 1;
        int y = Y[i] - 1;
        
        if (x < 0 || n <= x || y < 0 || n <= y) {
            ans++;
            continue;
        }
        
        if (t == 1) {
            if (same(x, y + n) || same(x, y + 2 * n)) {
                ans++;
            }
            else {
                unite(x, y);
                unite(x + n, y + n);
                unite(x + 2 * n, y + 2 * n);
            }
        }
        else {
            if (same(x, y) || same(x, y + 2 * n)) {
                ans++;
            }
            else {
                unite(x, y + n);
                unite(x + n, y + 2 * n);
                unite(x + 2 * n, y);
            }
        }
    }
    
    printf("%d\n", ans);
    
    return 0;
}