poj 1182 食物链

http://poj.org/problem?id=1182

并查集的灵活应用 

代码：

/*

可以这样理解

并查集是由很多树组成的，这些树不断的合并

下面代码f[]仍然代表此节点所属的树根

而d[]表示此点到父节点的差值，但是每次求fx()（树根）时，节点都更新指向树根

假如比父节点大1 则此点可以吃掉父节点

假如比父节点大2 则此点可以被父节点吃掉

假如和父节点相等，则属于同类

这里所指的大是针对当前节点对父节点而言的，2比1大1   0比2大1（循环）

由于更新完都指向树根了，所以就有了相同的参照物

根据不同的点和树根的关系，可以推算出一棵树任意两点之间的关系



问题就在于如何维护树的合并，f[]还是按照原来的方法，d[]需要根据实际情况取余

理解是注意自己画图

-----见代码注释



*/

#include<cstdio>



using namespace std;

const int N=50005;

int f[N],d[N];

int fx(int x)

{

    if(f[x]!=x)

    {

        int tmp=f[x];

        f[x]=fx(f[x]);

        //更新到这来时，x的父节点已经更新完成 指向根节点

        //而且x的父节点和根节点直接的关系也更新完成

        //结下了要更新x节点 因为x节点已经也指向了根节点（通过给f[x]赋值）

        //所以要根据父节点tmp到根节点的关系和x节点到父节点的关系

        //推算出x节点到根节点的关系，下面的式子就是计算过程 注意对3取余

        d[x]=(d[x]+d[tmp])%3;

    }

    return f[x];

}

int main()

{

    //freopen("data.in","r",stdin);

    int n,m;

    scanf("%d %d",&n,&m);

    for(int i=1;i<=n;++i)

    {f[i]=i;d[i]=0;}//初始化d[i]=0 因为每个节点开始指向自己 和自己是同类

    int ans=0;

    while(m--)

    {

        int a,b,w,A,B;

        scanf("%d %d %d",&w,&a,&b);

        if(a>n||b>n||a<=0||b<=0)

        {++ans;continue;}

        A=fx(a);

        B=fx(b);

        //这时候通过 fx(a),fx(b) a和b都更新完成，d[a]和d[b]都表示和各自根节点（可能相同，可能不同）关系

        if(w==1)

        {

            if(A==B)//假如 a和b 属于同一颗树

            {

                if(d[a]!=d[b])//如果不是同一类，矛盾

                ++ans;

            }else//假如不是同一颗树

            {

                f[A]=B;//合并两棵树

                //必须保证合并后d[a]和d[b]和树根B的关系一样

                //那么就得满足d[a]+d[A]==d[b]（%3） 所以。。。

                d[A]=(d[b]+3-d[a])%3;

            }

        }else

        {

            if(A==B)//假如 a和b 属于同一颗树

            {

                if((d[b]+1)%3!=d[a])//则d[a] 比 d[b] 大1 否则矛盾

                ++ans;

            }else//假如不是同一颗树

            {

                f[A]=B;//合并

                //同样的原理 只不过要多加1 因为a吃b

                d[A]=(d[b]+4-d[a])%3;

            }

        }



    }

    printf("%d\n",ans);

    return 0;

}