[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 ;

 }

秒客网

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

相关文章