HDU3635 Dragon Balls(并查集)

传送门:http://acm.hdu.edu.cn/showproblem.php?pid=3635
题意：有T组测试数据，每组数据输入n和m，表示有n个城市和m个查询。每个城市一开始的时候有一个龙珠，龙珠的编号与城市的编号相同。查询有两种。
T a b 表示将a龙珠当前所在城市内的所有龙珠转移到b城市。
Q a 表示查询a龙珠。输出三个数字。X（a龙珠当前在的城市）,Y（这个城市内的龙珠数量）,Z（a龙珠一共被移动了多少次）。

思路：这题看查询的问题，想到并查集的话，很容易就能想到怎么做。因为转移的时候是把这个城市内所有的龙珠都转移，所以只要并查集维护一下每个龙珠所在的位置，顺便维护一下块的大小，就可以解决每个询问的前两个问题。第三个问题是，一共被移动了多少次，那么我们在移动的时候，就用一个数组来记录每个城市的移动次数，比如a移动到b，就cnt[a]++。我们可以在合并的时候不压缩路径，就可以知道他曾经被移动到哪些城市，那么就可以从一开始的城市出发，一路找下去，累加路径上每个城市的移动次数，就是最终答案了。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <ctime>
using namespace std;
typedef long long LL;
typedef pair<int,int> pii;
#define PB push_back
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define calm (l+r)>>1
const int INF=1e9+7;

const int maxn=10000;
int pre[maxn+10],n,m;
int cnt[maxn+10];
int find(int x){
    int s;
    for(s=x;pre[s]>=0;s=pre[s]);
    return s;
}
void join(int a,int b){
    a=find(a);b=find(b);
    if(a==b)return;
    int sum=pre[a]+pre[b];
    cnt[a]++;
    pre[a]=b;pre[b]=sum;
}
int trans(int x){
    int ans=0;
    while(pre[x]>=0){
        //printf("trans:%d\n",x);
        ans+=cnt[x];
        x=pre[x];
    }
    return ans;
}
int main(){
    //freopen("D://input.txt","r",stdin);
    int T;scanf("%d",&T);
    int kase=1;
    while(T--){
        printf("Case %d:\n",kase++);
        scanf("%d%d",&n,&m);
        memset(pre,-1,sizeof pre);
        memset(cnt,0,sizeof cnt);
        char op[2];int a,b;
        while(m--){
            scanf("%s",op);
            if(op[0]=='T'){
                scanf("%d%d",&a,&b);
                join(a,b);
            }
            else{
                scanf("%d",&a);
                int x=find(a);
                printf("%d %d %d\n",find(a),-pre[x],trans(a));
            }
        }
    }
    return 0;
}