[HDOJ3635]Dragon Balls（并查集，路径压缩）

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

题意：有n个龙珠，n个城市。初始状态第i个龙珠在第i个城市里。接下来有两个操作：

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

Q A：查询A龙珠，要求：A龙珠所在城市，该城市龙珠数量，A移动的次数。

思路：用并查集可以轻松解决Q的前两个问题。关键在于如何统计A的移动次数，因为在T的时候是要将A所在城市的所有龙珠都要转移到B，那就要A集合里所有龙珠的移动次数都+1。假如我们在每一次转移前都去遍历一遍集合，显然是会TLE的，于是想了一个懒惰更新的方法。就是每次移动的时候只更新当前根节点的移动次数+1，合并的时候并在另一个树的根节点上。下一次移动这一个整棵树的时候必然要路径压缩，在路径压缩的时候将之前未更新的值更新即可。

 /*
 ━━━━━┒ギリギリ♂ eye！
 ┓┏┓┏┓┃キリキリ♂ mind！
 ┛┗┛┗┛┃＼○／
 ┓┏┓┏┓┃ /
 ┛┗┛┗┛┃ノ)
 ┓┏┓┏┓┃
 ┛┗┛┗┛┃
 ┓┏┓┏┓┃
 ┛┗┛┗┛┃
 ┓┏┓┏┓┃
 ┛┗┛┗┛┃
 ┓┏┓┏┓┃
 ┃┃┃┃┃┃
 ┻┻┻┻┻┻
 */
 #include <algorithm>
 #include <iostream>
 #include <iomanip>
 #include <cstring>
 #include <climits>
 #include <complex>
 #include <fstream>
 #include <cassert>
 #include <cstdio>
 #include <bitset>
 #include <vector>
 #include <deque>
 #include <queue>
 #include <stack>
 #include <ctime>
 #include <set>
 #include <map>
 #include <cmath>
 using namespace std;
 #define fr first
 #define sc second
 #define cl clear
 #define BUG puts("here!!!")
 #define W(a) while(a--)
 #define pb(a) push_back(a)
 #define Rint(a) scanf("%d", &a)
 #define Rll(a) scanf("%lld", &a)
 #define Rs(a) scanf("%s", a)
 #define Cin(a) cin >> a
 #define FRead() freopen("in", "r", stdin)
 #define FWrite() freopen("out", "w", stdout)
 #define Rep(i, len) for(int i = 0; i < (len); i++)
 #define For(i, a, len) for(int i = (a); i < (len); i++)
 #define Cls(a) memset((a), 0, sizeof(a))
 #define Clr(a, x) memset((a), (x), sizeof(a))
 #define Full(a) memset((a), 0x7f7f, sizeof(a))
 #define lp p << 1
 #define rp p << 1 | 1
 #define pi 3.14159265359
 #define RT return
 #define lowbit(x) x & (-x)
 #define onenum(x) __builtin_popcount(x)
 typedef long long LL;
 typedef long double LD;
 typedef unsigned long long ULL;
 typedef pair<int, int> pii;
 typedef pair<string, int> psi;
 typedef map<string, int> msi;
 typedef vector<int> vi;
 typedef vector<LL> vl;
 typedef vector<vl> vvl;
 typedef vector<bool> vb;

 const int maxn = ;
 int n, q;
 int pre[maxn];
 int t[maxn], c[maxn];

 int find(int x) {
     if(pre[x] == x) RT x;
     int px = pre[x];
     pre[x] = find(pre[x]);
     t[x] += t[px];
     RT pre[x];
 }

 void unite(int x, int y) {
     int fx = find(x);
     int fy = find(y);
     if(fx != fy) {
         pre[fx] = fy;
         t[fx]++;
         c[fy] += c[fx]; c[fx] = ;
     }
 }

 int main() {
     // FRead();
     int T, _ = ;
     int a, b;
     char cmd[];
     Rint(T);
     W(T) {
         Rint(n); Rint(q);
         Cls(t);
         Rep(i, n+) c[i] = , pre[i] = i;
         printf("Case %d:\n", _++);
         W(q) {
             Rs(cmd);
             if(cmd[] == 'T') {
                 Rint(a); Rint(b);
                 unite(a, b);
             }
             if(cmd[] == 'Q') {
                 Rint(a);
                 int fa = find(a);
                 printf("%d %d %d\n", fa, c[fa], t[a]);
             }
         }
     }
     RT ;
 }