POJ 1182 食物链(种类并查集)

题目分析

这道题主要有三种关系，就是A吃B，A被B吃，还有就是A与B是相同类型的。因此我们可以用一个数组re[]来表示与父根之间的关系，当值为0时这个表明当前该动物与父根动物是同一类型，如果为1那么该动物被父根动物吃，如果为2表明该动物可以吃父根动物，因此依赖关系就建立起来了。
对于每次询问如果这2个动物在一个并查集中，那么只需要判断一下即可，如果不在同一个并查集中那么需要建立关系。具体开代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 50010;

int fa[maxn], re[maxn];

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

    int temp = fa[x];  //找到父亲节点
    fa[x] = Find(fa[x]);
    re[x] = (re[x] + re[temp])%3; //那么当前节点与根节点的关系通过这个式子一步步往上推即可得到

    return fa[x];
}

void unite(int x, int y, int d){
    int fx = Find(x);
    int fy = Find(y);

    fa[fy] = fx;
    //y对x的关系是d-1，x对fx的关系为re[x],y对fy的关系是re[y](则fy对y的关系是(3-re[y])),故推得下式
    re[fy] = (re[x] - re[y] + 3 + (d-1))%3;  
}

void init(){
    for(int i = 0; i < maxn; i++)
        fa[i] = i, re[i] = 0;
}

int main(){
    int N, K, D, X, Y;
    scanf("%d %d", &N, &K);
    init();
    int ans = 0;
    while(K--){
        scanf("%d%d%d", &D, &X, &Y);
        if(X > N || Y > N || (D == 2 && X == Y)){ ans++; continue;}
        if(Find(X) == Find(Y)){
            if(D == 1){
                if(re[X] != re[Y]) ans++;
            }
            else if(D == 2){
                if((re[X]+1)%3 != re[Y]) ans++;  //这一步很好想，自己手动推一下即可
            }
        }
        else unite(X, Y, D);
    }
    printf("%d\n", ans);
    return 0;
}