hdu 2818 Building Block 并查集(边搜索边更新)

#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=3e4+10;
int f[maxn],c[maxn],num[maxn];
int find(int x)
{
    if(x!=f[x])
    {
        int t=find(f[x]);
        c[x]+=c[f[x]];
        f[x]=t;
    }
    return f[x];
}
void init()
{
    int i,j,k;
    for(i=0;i<maxn;i++)//需要初始化0,。。
    {
        f[i]=i;
        num[i]=1;
        c[i]=0;
    }
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        int i,j,k,a,b,aa,bb;
        char str[2];
        init();
        for(i=0;i<n;i++)
        {
            scanf("%s",str);
            if(str[0]=='M')
            {
                scanf("%d%d",&a,&b);
                aa=find(a);
                bb=find(b);
                if(aa!=bb)
                {
                    c[aa]=num[bb];
                    num[bb]+=num[aa];
                    f[aa]=bb;
                }
            }
            else
            {
                scanf("%d",&a);
                find(a);
                printf("%d\n",c[a]);
            }
        }
    }
    return 0;
}
/*
    并查集
    num[i]表示i所在堆的箱子个数。
    c[i]表示i箱子下面的箱子个数。
    通过并查集查找时更新数据，减少时间消耗，一堆箱子中，以最底层箱子为父节点。
    
    特别注意，这题箱子编号有0，。。。
*/