Poj-1182 食物链

[题目链接]

感悟：很经典的一道并查集题，做两遍了，收获满满的~~（强烈建议只看一点点思路）

思路：对每个节点维护其到根节点的偏移量（偏移量代表种类关系）。

• par[x]表示x的根节点，rel[x]表示x与根节点的偏移量。
• rel[x]=0表示x与根节点为同类，1表示根节点吃x，2代表x吃根节点。
• 同类/不同类时，如何判断关系是否矛盾以及如何合并两顶点所在的集合（真的值得自己想好几天）。

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int Max_n=5e4+10;

int n,m;
int par[Max_n];
int rel[Max_n];

int Find(int x){ //将沿途所有结点都指向根结点，并分别更新它们与根结点的关系
    if(par[x]==x)return x;
    else {
        int p=Find(par[x]); //p为x的根节点
        rel[x]=(rel[x]+rel[par[x]])%3;
        return par[x]=p;
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<=n;i++){ //初始化
        par[i]=i;
        rel[i]=0;
    }
    int k,x,y,ans=0;
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&k,&x,&y);
        if(x>n||y>n||(k==2&&x==y)){
            ans++;continue;
        }
        int fx=Find(x),fy=Find(y);
        if(fx==fy){ //根节点相同情况判断是否冲突
            if(k==1&&rel[x]!=rel[y])ans++;
            if(k==2&&rel[x]!=(rel[y]+2)%3)ans++;
        }
        else { //根节点不相同情况进行合并
            par[fy]=fx;
            if(k==1)rel[fy]=(rel[x]-rel[y]+3)%3;
            else rel[fy]=(1+rel[x]-rel[y]+3)%3;
        }
    }
    printf("%d\n",ans);
    return 0;
}