poj 1988 - Cube Stacking(并查集)

题意很简单，

思路：

         首先我们肯定得必须维护以为记录节点信息的数组cnt[]来记录该cube下面的所有cube的个数。

         由于对move操作，要更新放到上面的集合【节点a所在的集合】的所有节点的cnt信息，然而由于并查集特点，我们知道需要把这个集合放到并查集的子树上才方便更新信息。所有我们在做并查集合并的时候要把集合a并到集合b上。

         但是这样还不够，在合并的时候我们还需要上面集合中的所有节点的个数，而次集合cnt维护的只是它下面的cube的数量，所以我们还需要一个数组来维护各个集合的节点个数。

代码如下：

const int M = 30005;
int p[M], cnt[M], num[M];
char get_char()
{
    char tmp;
    while(scanf("%c", &tmp))
        if(tmp=='M'||tmp=='C') return tmp;
}
int find(int x)
{
    int tmp = p[x];
    p[x] = (p[x]==x?x:find(p[x]));
    cnt[x]+=cnt[tmp];
    return p[x];
}
int main()
{
    int P, a, b;
    char o;
    scanf("%d", &P);
    for(int i = 1; i < M; ++i) p[i] = i, num[i] = 1;
    for(int i = 0; i < P; ++i)
    {
        o = get_char();
        if(o=='C')
        {
            scanf("%d", &a);
            find(a);
            printf("%d\n", cnt[a]);
        }
        else
        {
            scanf("%d%d", &a, &b);
            int x = find(a);
            int y = find(b);
            p[x] = y;
            cnt[x] += num[y];
            num[y] += num[x];
        }
    }
    return 0;
}