食物链

题目

https://www.luogu.org/problemnew/show/P2024

思路

就是用3倍的并查积的存各种动物的关系

一倍存本身，二倍存猎物，三倍存天敌

唯一容易忽略的点就是：一的猎物的猎物 就是一的天敌

那么我们每次只要维护三个并查积的关系就可以了

代码

#include<iostream>
#include<cstdio>
using namespace std;
int f[50005],d[50005],n,k,d1,x,y,ans;
int find(int x)//寻找祖先的函数
{
    if(x!=f[x])
    {
        int xx=f[x];
        f[x]=find(f[x]);
        d[x]=(d[x]+d[xx])%3;
    }
    return f[x];
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {f[i]=i;d[i]=0;}
    for(int i=1;i<=k;i++)
    {
        scanf("%d%d%d",&d1,&x,&y);
        if((d1==2&&x==y)||(x>n||y>n))
        {ans++;continue;}
        if(d1==1)
        {
            if(find(x)==find(y))
            {if(d[x]!=d[y]) ans++;}
            else
            {
                d[f[x]]=(d[y]-d[x]+3)%3;
                f[f[x]]=f[y];
            }
        }
        if(d1==2)
        {
            if(find(x)==find(y))
            {if(d[x]!=(d[y]+1)%3) ans++;}
            else
            {
                d[f[x]]=(d[y]-d[x]+4)%3;
                f[f[x]]=f[y];
            }
        }
    }
    printf("%d\n",ans);
}