poj1988 Cube Stacking

并查集的高效之处在于路径压缩和延迟更新。

在本题中需要额外维护子树的规模以及当前子树节点到跟的距离两个数组。

由于一个新的数必然是两棵树拼接而成，对于子树规模的更新直接相加即可，

对于节点到跟的距离：

我们让a树的根指向b树的根，同时更新a树根到(a+b)树根（即b树根）的距离为size_of(Tree(b))即可完成维护。

 

http://poj.org/problem?id=1988

 

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 1e5 + 10;
int m, a, b;
int fa[maxn], d[maxn], size[maxn];
char ch;

int father(int u){
    //path compressing
    if(fa[u] == u) return u;
    int t = fa[u];
    fa[u] = father(fa[u]);
    d[u] += d[t];
    return fa[u];
}

void solve(){
    if(ch == 'M'){
        int ra = father(a);
        int rb = father(b);
        fa[ra] = rb;
        d[ra] += size[rb];
        size[rb] += size[ra];
        return;
    }
    int ra = father(a);
    printf("%d\n", d[a]);
}

int main(){
    #ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    #endif
    while(~scanf("%d", &m)){
        for(int i = 0; i <= m; i++) fa[i] = i, d[i] = 0, size[i] = 1;
        for(int i = 0; i < m; i++){
            scanf(" %c%d", &ch, &a);
            if(ch == 'M') scanf("%d", &b);
            solve();
        }
    }
    return 0;
}