hdu 3635 Dragon Balls (带权并查集)

题目：

        链接：点击打开链接

题意：

        把编号为1~n的龙珠放到编号为1~n的城市中去。现在有T和Q两种操作：

        T：A B 把A龙珠所在城市的所有龙珠转移到B城市中。

        Q：A 表示查询A龙珠的一些信息：X（A所在的城市），Y（A所在城市龙珠的数目），Z（A转移到该城市被移动的次数）。

思路：

        

代码：

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

const int N = 10010;

struct node{
    int parent;
    int son;
    int TransNum;
}p[N];

int n,q;
int a,b,c,ans;
char ch;

int findset(int x)
{
    int temp;
    if(x == p[x].parent)
        return x;
    else
    {
        temp = p[x].parent;
        p[x].parent = findset(p[x].parent);
        p[x].TransNum += p[temp].TransNum;
    }
    return p[x].parent;
}

void mergeset(int x,int y)
{
    int fx = findset(x);
    int fy = findset(y);
    if(fx != fy)
    {
        p[fx].parent = fy;
        p[fx].TransNum++;
        p[fy].son += p[fx].son;
        p[fx].son = 0;
    }
}

void init()
{
    for(int i=1; i<=n; i++)
    {
        p[i].parent = i;
        p[i].son = 1;
        p[i].TransNum = 0;
    }
}

int main()
{
    //freopen("input.txt","r",stdin);
    int t;
    int kase = 0;
    cin>>t;
    while(t--)
    {
        scanf("%d%d",&n,&q);
        getchar();
        init();
        printf("Case %d:\n",++kase);
        for(int i=0; i<q; i++)
        {
            scanf("%c",&ch);
            getchar();
            if(ch == 'T')
            {
                scanf("%d%d",&a,&b);
                mergeset(a,b);
            }
            else
            {
                scanf("%d",&c);
                ans = findset(c);
                printf("%d %d %d\n",ans,p[ans].son,p[c].TransNum);
            }
            getchar();
        }
    }
    return 0;
}

----------------------------------------------

战斗，从不退缩；奋斗，永不停歇~~~~~~~~