poj-1182（带权并查集）_codestorm_新浪博客

转自： http://blog.sina.com.cn/s/blog_88705ca20100ugc2.html

算法思路：把那些确定了相对关系的节点放在同一棵树里（可以同时存在多棵树，单独每棵树中节点的相对关系确定），每个节点对应的 v[] 值记录他与父节点的关系（ 0:同类； 1:根吃他； 2:他吃根 ）。当不在同一棵树上时，只需把 aa，bb 两个根节点连接，并修改变成aa子结点的bb相对于aa的关系。这里 根据 a ，b分别对应于aa ，bb的关系（即：v[a] ,v[b]）就可以确定 bb相对于新根aa的关系。v[bb] = (v[a]-v[b]+3+d)%3。v[b]代表b作为子结点与bb的关系，反过来bb作为子结点与b的关系则为-v[b]，加3是为了防止出现负数。

#include

#include

#include

#include

using namespace std;

int front[50005],v[50005];

int find(int x)

{

    if(x!=front[x])

       {

             int y=front[x];

               front[x]=find(front[x]);

              v[x]=(v[x]+v[y])%3;      //回溯中由子结点与父亲结点的关系和父亲结点与根结点的关系找到该子结点与根结点的关系

        }

     else return x;

        return front[x];

}

void link(int x,int y,int z)

{

      int xx=find(x),yy=find(y);

    front[yy]=xx;

 v[yy]=(v[x]-v[y]+3+z)%3;  //由子结点与根结点的关系来找到根结点与根结点的关系

}

int main()

{

   int n,m,i,count=0;

    memset(v,0,sizeof(v));

        for(i=0;i<50005;i++)front[i]=i;

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

  while(m--)

    {

             int x,y,z;

            scanf("%d%d%d",&z,&x,&y);

             if(x>n||y>n)count+=1;

           else

          {

                     if(z==1)

                      {

                             if(find(x)!=find(y))link(x,y,0);

                              else if(v[x]!=v[y])count+=1;     //同类则在同一集合，且与根结点的关系相同。

                     }

                     else

                  {

                             if(find(x)!=find(y))link(x,y,1);

                              else if((v[x]+1)%3!=v[y])count+=1;   //在同一根结点，如果两子结点是捕食关系则它们与根结点的关系值差1

                     }

             }

     }

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

 return 0;

}