Cube Stacking

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

题意：
给定N个方块，排成一行，将它们编号1到N。再给出P个操作：
①M i j表示将i所在的那一堆移到j所在那一堆的顶上。
②C i表示一个询问，询问i下面有多少个方块。

思路：
带权并查集，一堆中最顶上的方块作为父节点，用dis[X] 统计X到父亲节点的距离，num[fa[X]]表示这个集合的大小，两者相减即为答案。

标程：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<string>
#include<set>
#include<map>
#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;

#define ll long long
const int inf = 0x3f3f3f3f;
const int maxn = 300010;
const double pi = acos(-1.0);

int pre[maxn],num[maxn],dis[maxn];
int n;

void init()
{
    for(int i=0; i<n; i++)
    {
        pre[i]=i;
        num[i]=1;
        dis[i]=0;
    }
}
int Find(int x)
{
    if(x!=pre[x])
    {
        int t=pre[x];
        pre[x]=Find(pre[x]);
        dis[x]+=dis[t];
    }
    return pre[x];
}

int main()
{
    while(~scanf("%d",&n))
    {
        init();
        char op;
        while(n--)
        {
            scanf(" %c",&op);
            int x,y;
            if(op=='M')
            {
                scanf("%d %d",&x,&y);
                int fx=Find(x);
                int fy=Find(y);
                if(fx!=fy)
                {
                    pre[fy]=fx;
                    dis[fy]=num[fx];
                    num[fx]+=num[fy];
                }
            }
            else
            {
                scanf("%d",&x);
                int fx=Find(x);
                printf("%d\n",num[fx]-dis[x]-1);
            }
        }
    }
    return 0;
}