POJ 1988 Cube Stacking (带权并查集)

题目链接：http://poj.org/problem?id=1988

有n个元素，开始每个元素自己 一栈，有两种操作，将含有元素x的栈放在含有y的栈的顶端，合并为一个栈。第二种操作是询问含有x元素下面有多少个元素。

经典的带权并查集，cnt表示包含这个元素的集合中所有元素个数，dis表示这个元素离最上面元素的个数(距离)。

看代码领会一下吧。

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
int par[int(3e4 + 5)] , cnt[int(3e4 + 5)] , dis[int(3e4 + 5)];
//cnt[i]表示i所在点的大小，dis[i]表示i离最上面节点的距离(个数)
int Find(int n) {
    if(par[n] == n)
        return n;
    int temp = Find(par[n]); //先算出n的父节点的dis
    dis[n] += dis[par[n]];
    par[n] = temp; //路径压缩
    return temp;
}

void Union(int u , int v) {
    int fu = Find(u) , fv = Find(v);
    if(fu == fv)
        return ;
    par[fv] = fu; 
    dis[fv] = cnt[fu]; //Find函数中dis[fv]并没有回溯增加过
    cnt[fu] += cnt[fv]; //总个数相加
}

int main()
{
    int n , u , v;
    char q[3];
    while(~scanf("%d" , &n)) {
        memset(dis , 0 , sizeof(dis));
        for(int i = 1 ; i <= 3e4 ; ++i) {
            cnt[i] = 1;
            par[i] = i;
        }
        for(int i = 0 ; i < n ; ++i) {
            scanf("%s" , q);
            if(q[0] == 'M') {
                scanf("%d %d" , &u , &v);
                Union(u , v);        
            }
            else {
                scanf("%d" , &u);
                int x = Find(u); //回溯累加一次
                printf("%d
" , cnt[x] - dis[u] - 1);
            }
        }
    }
    return 0;
}