并查集———食物链

POJ 食物链

题目链接

这里是输出假话的数目
三种情况
1）x或y大于n
2)自己吃自己
3）与前面的话矛盾

该怎么表示每个元素之间的关系，这是一个难点

这里的办法是

将每个元素关系放在一棵树中
通过x和y两者与根节点的关系来判断语句的正确与否

这里有三种种类的生物，即以3为一个周期
用并查集来维护他们之间的关系

如果d[x]>d[y]表示x吃y；
如果d[x]=d[y]则是同类；

用他们与根节点的距离来表示他们的关系
利用并查集来维护距离

#include<iostream>
#include<cstdio>
using namespace std;
const int N=50010;
int a[N],d[N],ans;
int find(int x)
{
    if(x!=a[x])
    {
        int t=find(a[x]);
        //这里是x的祖宗结点
        d[x]+=d[a[x]];
        //x到根节点的距离就是
        //x到父节点的距离加上父节点到祖宗结点的距离
        a[x]=t;//路径压缩
    }
    return a[x];
}
int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        a[i]=i;
    //开始时每个元素的父节点都是自己本身
    while(k--)
    {
        int m,x,y;
        scanf("%d%d%d",&m,&x,&y);
        if(x>n||y>n) ans++;
        //第一种假话情况
        else
        {
            int px=find(x),py=find(y);
        //判断两点是否在一棵树上
            if(m==1)
            {
                if(px==py&&(d[x]-d[y])%3) ans++;
        //在一棵树上时直接判断，如果他们和根节点距离相等则是同类
                else if(px!=py)
                {
                    a[px]=py;
                    d[px]=d[y]-d[x];
                }
        //不在同一棵树上时合并在一棵树上，
        //保证他们两个到根节点距离相等
        //不在一棵树上表示这两个之前没有关系，因此是真话
                
            }
            else
            {
                if(px==py&&(d[x]-d[y]-1)%3) ans++;
                else if(px!=py)
                {
                    a[px]=py;
                    d[px]=d[y]+1-d[x];
        //因为是x吃y,这里保证x到根节点的距离比y到根节点的距离多1
                }
            }
        }
    }
    printf("%d",ans);
    return 0;
}

有些oj用c++的标准输入输出流可能会超时