poj1182食物链 并查集

    这题与hdu1829做法类似，在一棵树中，用0表示该节点与根节点相同，1表示该节点被根节点吃，-1表示该节点吃根节点。

#include<iostream>#include<cstdio>using namespace std;const int maxn=50005;int set[maxn],c[maxn];//set记录父亲节点,c记录当前节点与根节点的关系//0 ：同类 1：被根节点吃 -1：吃根节点void init(int n){    for(int i=1;i<=n;i++)    {        set[i]=i;        c[i]=0;    }}inline int change(int x){    if(x==2)        return -1;    else if(x==-2)        return 1;    else        return x;}int find(int x){    if(set[x]==x)        return x;    int t=set[x];    set[x]=find(t);    c[x]=change(c[x]+c[t]);//路径压缩并更新节点与根节点的关系    return set[x];}bool merge(int x,int y,int d){    int f1=find(x);    int f2=find(y);    if(f1==f2)        return false;    set[f1]=f2;    c[f1]=change((c[y]-d-c[x]+1)%3);//画出食物链可以看出来    return true;}int main(){    int n,m;    int d,a,b,sum=0;    scanf("%d%d",&n,&m);    init(n);    while(m--)    {        scanf("%d%d%d",&d,&a,&b);        if(d==2&&a==b||a>n||b>n)        {            sum++;            continue;        }        //若两种动物处于同一条食物链且不符合当前关系，则为假话        if(!merge(a,b,d)&&change(c[b]-c[a])!=d-1)            sum++;    }    printf("%d\n",sum);    return 0;}