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 = ;
 int par[maxn], h[maxn], ch[maxn];
 void init()
 {
     for(int i = ; i < maxn; i++) {
         par[i] = i;
         h[i] = ;
         ch[i] = ;
     }
 }
 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[];
             scanf("%s", s);
             if(s[] == 'M') {
                 int a, b;
                 scanf("%d%d", &a, &b);
                 Union(a, b);
             }
             else {
                 int a; scanf("%d", &a);
                 Find(a);
                 printf("%d\n", h[a]);
             }
         }
     }
     return ;
 }