食物链 POJ - 1182 (带权并查集)

题目链接

这个题给的都是相对关系，不能够确定具体的某个动物属于哪个物种，因此有开三倍空间来处理的那样的思路做法①链接

带权并查集就是用来处理相对关系的，所以也可以考虑用带权并查集来做。

图中是三种动物的相对关系,x->y表示x吃y

我们用数组v[i]表示i与父节点的相对关系:
v[i]=1 表示i 吃 fa[ i ]
v[i]=2 表示i 被fa[ i ]吃
v[i]=0 表示i fa[ i ]是同类

路径压缩时： v[x ]=（v[x]+v[ fa[x] ]）%3

合并和判断矛盾见代码

#include<iostream>
#include<cstdio>
using namespace std;
#define ll long long
const int maxn = 5e5+7;
const int INF = 1e9+7;
//有三种动物 共n只 满足a吃b b吃c c吃a   给出m句话描述关系 输出假话数目
//思路：带权值的并查集
int n,m;
int fa[maxn];
int v[maxn];//0代表同类 1表示吃 2表示被吃
int find(int x)
{
    if(fa[x] == x) return x;
    int f=fa[x];
    fa[x]=find(fa[x]);
    v[x]=(v[x]+v[f])%3;//路径压缩
    return fa[x];
}
bool Union(int x,int y,int k)
{
    int r1=find(x),r2=find(y);
    if(r1 == r2)//之前存在关系 这种情况下可能是假话
    {
        int w=(v[x]-v[y]+3)%3;//得到x y的相对关系
        if(w==k) return 1;//真话
        else return 0;
    }
    fa[r1]=r2;
    v[r1]=(v[y]+k-v[x]+3)%3;//合并 可以画图理解，一个矩形
    return 1;
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++) fa[i]=i;
    int ans=0;
    while(m--)
    {
        int k,x,y;
        scanf("%d %d %d",&k,&x,&y);
        if(x<1 || x>n || y<1 || y>n ) 
        {
            ans++;
            continue;
        }
        if(!Union(x,y,k-1)) ans++;//注意是k-1  因为我们让0表示同类 1表示吃
    }
    cout<<ans<<'\n';
    return 0;
}