[HDU 3635]Dragon Balls[并查集][路径压缩]

题目链接： [HDU 3635]Dragon Balls[并查集][路径压缩]

题意分析：

'T‘操作，代表将A龙珠所在城市的所有龙珠转移到B龙珠所在的城市。

’Q‘操作，查询A龙珠所在的城市，该城市的龙珠数和A龙珠总共转移了多少次。

解题思路：

我们使用并查集，将在同一个城市的龙珠并在一起。这样查询所在城市就So easy了。

剩下的问题就是城市的龙珠数和转移次数。

龙珠数，我们可以使用一个in[]数组，在城市合并时，比如A->B，in[B] += in[A]即可。

转移次数的话，如果一个龙珠已经合并在父亲下，那么父亲移动的一次，这个龙珠移动的次数也会相应加一次。由这个道理：我们可以在路径压缩的时候合并转移次数。

个人感受：

刚开始二话不说写了发搜索= =。MLE，即使不MLE，估计也TLE。然后队友说用并查集= =，其实我也想到了，可是就是不知道怎么处理龙珠的转移次数，当时应该好好再想想怎么用并查集表示转移的TAT

具体代码如下：

#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iomanip>
#include<iostream>
#include<map>
#include<queue>
#include<set>
#include<sstream>
#include<stack>
#include<string>
#define ll long long
#define pr(x) cout << #x << " = " << (x) << '\n';
using namespace std;

const int INF = 0x7f7f7f7f;
const int MAXN = 1e4 + 111;

int mv[MAXN], p[MAXN], in[MAXN];

int find(int x) {
    if (x == p[x]) {
        return p[x];
    }
    int temp = p[x];
    p[x] = find(p[x]);
    mv[x] += mv[temp];
    return p[x];
}

int main()
{
    int t, n, q;
    for (int kase = scanf("%d", &t); kase <= t; ++kase) {
        scanf("%d%d", &n, &q);
        printf("Case %d:\n", kase);
        for (int i = 1; i <= n; ++i) {
            mv[i] = 0;
            in[i] = 1;
            p[i] = i;
        }
        int u, v;
        char op[2];
        while (q --) {
            scanf("%s %d", op, &u);
            if (op[0] == 'T') {
                scanf("%d", &v);
                u = find(u), v = find(v);
                if (u != v) {
                    ++mv[u];
                    p[u] = v;
                    in[v] += in[u];
                }
            }
            else {
                int x = u;
                u = find(u);
                printf("%d %d %d\n", u, in[u], mv[x]);
            }
        }
    }
    return 0;
}