hdu 2818(带权并查集）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2818

思路：Count[i]表示i下面的积木个数，路径压缩的时候更新一下即可，sum[i]表示以i为根的积木的个数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 30000+30
int parent[MAXN];
int sum[MAXN],Count[MAXN];
int n;

void Initiate() {
    memset(Count,0,(MAXN)*sizeof(int));
    for(int i=0; i<=MAXN; i++) {
        parent[i]=i;
        sum[i]=1;
    }
}

int Find(int x) {
    if(x==parent[x])
        return x;
    int tmp=Find(parent[x]);
    Count[x]+=Count[parent[x]];
    return parent[x]=tmp;
}

void Union(int u,int v) {
    int r1=Find(u),r2=Find(v);
    if(r1==r2)return ;
    parent[r1]=r2;
    Count[r1]=sum[r2];
    sum[r2]+=sum[r1];
}

int main() {
    char str[2];
    int u,v;
// freopen("1.txt","r",stdin);
    scanf("%d",&n);
    Initiate();
    while(n--) {
        scanf("%s",str);
        if(str[0]=='M') {
            scanf("%d%d",&u,&v);
            Union(u,v);
        } else {
            scanf("%d",&u);
            Find(u);
            printf("%d\n",Count[u]);
        }
    }
    return 0;
}