POJ 1988 Cube Stacking（带权并查集）

题意：

给定 n 个 方块， 然后有 p 个操作

操作M (a , b) ， 就是将a方块所在的那一堆方块放到 b 上面。

操作C （x） ， 就是询问x方块下面有多少方块

分析：

记录方块到父亲的距离dis， 还有最底的方块的cnt。

两个方块合并时，合并方块堆的父亲dis += 最底方块堆的cnt ，最底的方块堆的cnt 加上合并方块堆的 cnt

那么对于每个询问， 只要做一次路径压缩， 从根节点一直更新到询问点， 那么就能更新出询问点的长度

代码：

#include <bits/stdc++.h>
using namespace std;
const int Maxn =  10 ;
int P;
int F[Maxn], cnt[Maxn], dis[Maxn];
void init(){
    for(int i = 0; i < Maxn; i++)
        F[i] = i, cnt[i] = 1, dis[i] = 0;
}


int Find (int x){
    if (F[x] != x){
        int temp  = F[x]; //设置延时，最后从父节点回溯的时候再更新，注意与下面的对比
        F[x] = Find(F[x]);
        dis[x] += dis[temp];
        return F[x];
    }
    return x;
}

/* 不能写成以下的形式， 因为这样会先更新了自己， 然后再更新父节点
int Find (int x){
    if (F[x] != x){
        dis[x] += dis[F[x]];
        F[x] = Find(F[x]);
        return F[x];
    }
    return x;
}
*/


int Merge(int a, int b){
    int t1 = Find(a), t2 = Find(b);
    if(t1 != t2){
        F[t1] = t2; //将t1的根改为t2
        dis[t1] = cnt[t2]; //那么t1到根的距离就是cnt[t2]
        cnt[t2] += cnt[t1]; //然后cnt[t2]这一堆就要加上cnt[t1]
    }
}
int main(){
    freopen("debug.txt","r", stdin);
    init();
    scanf("%d", &P);
    while(P--)
    {

        char ch[9];
        scanf("%s", ch);
        if(ch[0] == 'M'){
            int a, b;
            scanf("%d %d", &a, &b);
            Merge(a,b);
        }
        else{
            int x;
            scanf("%d", &x);
            Find(x);
            printf("%d
", dis[x]);
        }
    }
    return 0;
}