Building Block HDU

Building Block

HDU - 2818

题意：搬砖。。。每一次可以把a所在的那一堆放到b所在的那一堆上面，问第x号砖下面有几块砖。

记录一下到根节点的距离(dw)，以及根节点上方有几块砖(up)。

#include <bits/stdc++.h>
using namespace std;

const int maxn=30010;
int f[maxn],up[maxn],dw[maxn];

int gf(int x){
    if(x==f[x]){
        return x;
    }
    int r=f[x];
    f[x]=gf(r);
    dw[x]+=dw[r];
    return f[x];
}

int n;
int main(){
   // freopen("in.txt","r",stdin);
    while(scanf("%d",&n)!=EOF){
        int ans=0;
        for(int i=0;i<maxn;i++) {f[i]=i;up[i]=1;dw[i]=0;}
        for(int i=0;i<n;i++){
            int a,b;
            char c;
            getchar();
            scanf("%c",&c);
            if(c=='M'){
                scanf("%d%d",&a,&b);
                int pa=gf(a),pb=gf(b);
                if(pa!=pb){
                    f[pa]=pb;
                    dw[pa]+=up[pb];
                    up[pb]+=up[pa];
                }
            }else{
                scanf("%d",&a);
                gf(a);
                printf("%d
",dw[a]);
            }
        }
    }
    return 0;
}