poj1988-种类并查集

题意：有N个立方体（1<=N <=30,000），相应地初始时每个立方体放在一个栈中，有两种操作：1、M X Y：把包含第X个立方体的栈中的所有立方体当做一个整体拿出来压入包含第Y个立方体的栈中（即是放在Y栈中的栈顶）；2、C X：询问编号为X的立方体下有几个立方体。

思路：把每次1操作后的栈底元素当做根节点。记录栈中元素到根节点的距离h[]，记录栈中元素的子节点个数ch[]。Find（）路径压缩的时候更新的当前节点到根节点的距离h[]即是h[]加上其原父节点到根节点的距离。合并两个栈元素的时候直接合并两根节点，设根节点为X 和 Y，Y 为新的根节点， X与Y的距离h[X]为X还没放入栈中栈里的元素个数，即

h[x] = ch[Y]，ch[Y] += ch[x]。

总结：开始自己想的是只记录更新除父节点数组之外的一个节点信息，然后栈顶元素当父节点，结果掉进了死胡同，后来考虑到栈中元素到栈底的距离才把思路掰回来了。有时候想不明白还是要多想想数据多画画图。

AC代码：

#include <cstdio>
#include <iostream>
#include <vector>
const int maxn = 30004;
int par[maxn], h[maxn], ch[maxn];
void init()
{
    for(int i = 0; i < maxn; i++) {
        par[i] = i;
        h[i] = 0;
        ch[i] = 1;
    }
}
int Find(int x)
{
    if(x == par[x]) return x;
    int t = par[x];
    par[x] = Find(par[x]);
    h[x] += h[t];
    return par[x];
}
void Union(int a, int b)
{
    int x = Find(a);
    int y = Find(b);
    if(x != y) {
        par[x] = y;
        h[x] = ch[y];
        ch[y] += ch[x];
    }
}

int main()
{
    int p;
    while(scanf("%d", &p) != EOF) {
        init();
        while(p--) {
            char s[2];
            scanf("%s", s);
            if(s[0] == 'M') {
                int a, b;
                scanf("%d%d", &a, &b);
                Union(a, b);
            }
            else {
                int a; scanf("%d", &a);
                Find(a);
                printf("%d
", h[a]);
            }
        }
    }
    return 0;
}