HDU 3635 Dragon Balls

http://acm.hdu.edu.cn/showproblem.php?pid=3635

并查集

改变一个点的值，同时要改变他儿子的值，用并查集实现

#include <cstdio>
#include <cstring>
using namespace std;

const int N=10010;
int set[N],cnt[N],sum[N];
int find(int x)
{
    if(x==set[x]) return x;
    int t=find(set[x]);
    cnt[x]+=cnt[set[x]];
    return set[x]=t;
}
int main()
{
    int T,C=0;
    scanf("%d",&T);
    while(T--)
    {
        printf("Case %d:\n",++C);
        int n,q;
        scanf("%d%d",&n,&q);
        memset(cnt,0,sizeof(cnt));
        for(int i=1;i<=n;i++) sum[i]=1;
        for(int i=1;i<=n;i++) set[i]=i;
        while(q--)
        {
            char op[2];
            scanf("%s",op);
            if(op[0]=='T')
            {
                int a,b;
                scanf("%d%d",&a,&b);
                int fa=find(a), fb=find(b);
                set[fa]=fb;
                sum[fb]+=sum[fa];
                cnt[fa]++;
            }
            else
            {
                int a;
                scanf("%d",&a);
                int fa=find(a);
                printf("%d %d %d\n",fa,sum[fa],cnt[a]);
            }
        }
    }
    return 0;
}