题目链接:
[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;
}