POJ 1988(带权并查集)

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

 

题目大意：最开始有N个砖头(1<=N<=30000)，然后进行p次操作，M x y操作是将带x编号的那堆石头放到带y编号的那堆石头的头上，C x操作是查询x编号下面有几块石头

 

题目思路：设置一个sum[x]表示x为首的那堆石头里有几个石头，dis[x]记录x到自己那堆石头最头上石头的距离。当M x y的时候，将带x的石头移到带y的石头那里，也就是现在x那堆石头的老大变成了y那堆石头的老大，y的老大到x老大的距离就变成了x那堆石头的个数，而x那堆石头的数量变成了原有的数量加上y那堆石头的数量。然后并查集的时候需要路径压缩。不断得通过自己的父亲往上找，然后把自己那一条链上的所有点的父亲都变成老大，并且把距离更新成到老大的距离。
 

以下是代码：

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 30005
#define inf 0x3f3f3f3f
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
int pre[MAXN],sum[MAXN],dis[MAXN];
char s[5];
int Find(int x){
    if(x==pre[x])return x;
    int root=Find(pre[x]);
    dis[x]+=dis[pre[x]];
    return pre[x]=root;
}
int main()
{
    int n,x,y;
    while(~scanf("%d",&n)){
        rep(i,1,MAXN){
            pre[i]=i;
            sum[i]=1;
            dis[i]=0;
        }
        rep(i,1,n){
            scanf("%s",s);
            if(s[0]=='M'){
                scanf("%d%d",&x,&y);
                int xx=Find(x),yy=Find(y);
                if(xx!=yy){
                    pre[yy]=xx;
                    dis[yy]=sum[xx];
                    sum[xx]+=sum[yy];
                }
            }
            else{
                scanf("%d",&x);
                int xx=Find(x);
                printf("%d\n",sum[xx]-dis[x]-1);
            }
        }
    }
    return 0;
}