poj 1182 (关系并查集) 食物链

题目传送门：http://poj.org/problem?id=1182

这是一道关系型并查集的题，对于每个动物来说，只有三种情况：同类，吃与被吃；

所以可以用0,1,2三个数字代表三种情况，在使用并查集的时候再多加一个关系数组，初始时全部赋值为0

然后就是在进行并查集的每一步时加入关系的改变，

如果祖先节点相同，说明两者之间的关系已经出现，是已知的，判断两者关系之和与给的d-1是否相同

祖先节点不同，说明在此之前两者的关系还未知，就赋予关系

噢 对了 poj 上的这题要单组输入，多组会wa

code

 

#include<cstdio>
using namespace std;
int father[50005];
int relat[50005];
void give (int n)
{
    for (int i=1;i<=n;i++)
    {
        father[i]=i;
        relat[i]=0; //初始关系都为0
    }
}
int find(int x)
{
    if (x==father[x]) return father[x];
    int t=find(father[x]);
    relat[x]=(relat[x]+relat[father[x]])%3; //在寻找祖先节点的过程中也要更新关系
    father[x]=t;
    return father[x];
}
int main()
{
    int n,k,sum,d,x,y,sx,sy;
    scanf("%d %d",&n,&k);
    give(n);
    sum=0;
    while (k--)
    {
        scanf("%d %d %d",&d,&x,&y);
        if (x>n||y>n)
        {
            sum++;
            continue;
        }
        if (d==2&&x==y) //不能自己吃自己啊
        {
            sum++;
            continue;
        }
        sx=find(x);
        sy=find(y);
        if (sx==sy) //说明关系可以确定，因为在之前出现过
        {
            if ((relat[x]-relat[y]+3)%3!=d-1)  //加三为了避免负数
              sum++;
        }
        else    //关系还不确定，那么不能判断，只能赋予关系
        {
            father[sx]=sy;
            relat[sx]=(relat[y]-relat[x]+3+d-1)%3;  //sx,sy,x,y的顺序不要反了，这都是对应的
        }
    }
    printf("%d\n",sum);
    return 0;
}