poj1988 Cube Stacking（并查集

题目地址：http://poj.org/problem?id=1988

题意：共n个数，p个操作。输入p。有两个操作M和C。M x y表示把x所在的栈放到y所在的栈上（比如M 2 6:[2 4]放到[1 6]上为[2 4 1 6]）,C x为输出x下面有几个数。

思路：并查集每个集合以栈最下面的数为根，维护两个数组num[x]表示x所在集合节点总数，count[x]表示x下方节点个数。每次查找压缩路径的时候更新count（换父节点的时候每轮都把父节点的count加给儿子，就可以一直更新到x所在栈的最底下），合并的时候更新px的count和py的num（把x的栈放到y的栈上，x下面多了num[y]个节点，新栈总根y总数增加num[x]个）。【不管是M还是C，先found一下x，把x的fa[x]更新到栈的最底下，count[x]才是真正的数！】

 #include <iostream>
 #include <cstdio>

 using namespace std;

 const int maxn =  + ;
 struct c
 {
     int fa, count, num;
 }cube[maxn];

 void init(int n)
 {
     for(int i = ; i <= n; i++)
     {
         cube[i].fa = i;
         cube[i].num = ;
         cube[i].count = ;
     }
 }

 int found(int x)
 {
     if(cube[x].fa == x)
         return x;
     int px = cube[x].fa;
     cube[x].fa = found(cube[x].fa);
     cube[x].count += cube[px].count;
     return cube[x].fa;
 }

 void unite(int x, int y)
 {
     int px = cube[x].fa;
     int py = cube[y].fa;
     if(px != py)
     {
         cube[px].fa = py;
         cube[px].count += cube[py].num;
         cube[py].num += cube[px].num;
     }
 }

 int main()
 {
     int p, x, y;
     char opr;
     init(maxn);
     scanf("%d", &p);
     while(p--)
     {
         getchar();
         scanf("%c", &opr);
         if(opr == 'M')
         {
             scanf("%d%d", &x, &y);
             int px = found(x);
             int py = found(y);
             if(px != py)
                 unite(x, y);
         }
         else
         {
             scanf("%d", &x);
             found(x);
             printf("%d\n", cube[x].count);
         }
     }
     return ;
 }