POJ 1988 Cube Stacking

* 1988 Cube Stacking
* 题意给你 1-n 编号的立方体，然后移动含有指定编号的立方体
* 栈移到另一个栈上边，问指定的编号立方体下面有多少个立方体。
*
* 应用 并查集的思路，移动一个栈时，相当于union_set操作，只要另开一个
* 数组记录立方体的位置，当合并时，只要改变根节点的位置记录就可以了，
* 这个地方并查集用的比较巧妙。其余的就是基本的并查集操作了。
* */

#include < iostream >
using   namespace  std;
int  cube[ 30330 ],nr[ 30330 ];
int  n;
void  make_set()
{
     for ( int  i = 0 ;i <= 30030 ; ++ i)
    {
        cube[i]  =   - 1 ;
        nr[i]  =   0 ;
    }
}
int  find_set( int  x)
{
     int  tmp  =  cube[x];
     if (cube[x] < 0 )
         return  x;
    cube[x]  =  find_set(cube[x]);
    nr[x]  +=  nr[tmp];
     return  cube[x];
}
void  union_set( const   int &  root1, const   int &  root2)
{
     int  tmp  =  cube[root1];
    cube[root1]  =  root2;
    nr[root1]  = nr[root1]  -  cube[root2];
    cube[root2]  +=  tmp;
}
int  main()
{
     int  i,j,root1,root2,n1,n2;
     char  ch[ 1 ];
    cin >> n;
    make_set();
     while (n -- )
    {
        scanf( " %s  " ,ch);    
         if (ch[ 0 ] == ' M ' )
        {
            scanf( " %d%d " , & n1, & n2);
             int  root1  =  find_set(n1);
             int  root2  =  find_set(n2);
             if (root1 != root2)
                union_set(root1,root2);
        } else   if (ch[ 0 ] == ' C ' )
        {
            scanf( " %d " , & n1);
             int  tmp  = find_set(n1);
            printf( " %d\n " ,nr[n1]);
        }
    }
     return   0 ;
}