hdu 3635 Dragon Balls 并查集应用记录每个点的转移次数

题意：

有n个龙珠，分别存放在编号为1-n的城市里面。然后进行如下操作，T,a,b将a龙珠所在城市的所有龙珠都转移到b龙珠所在的城市，Q,a输出a龙珠所在的城市，以及该城市一共有多少个龙珠，同时输出该龙珠被转移的次数，

思路：

并查集，前两项比较好像，后一项记录转移次数，我们只需要记录每一个点到达父节点所需的转移次数no[i]然后再路径压缩的时候求一个和，压缩完成后该点的no值就是该点的转移次数了。

View Code

#include <iostream>
#include <cstdio>
#include <cmath>
#include <vector>
#include <cstring>
#include <algorithm>
#include <string>
#include <set>
#include <functional>
#include <numeric>
#include <sstream>
#include <stack>
#include <map>
#include <queue>

#define CL(arr, val)    memset(arr, val, sizeof(arr))
#define lc l,m,rt<<1
#define rc m + 1,r,rt<<1|1
#define pi acos(-1.0)
#define ll __int64
#define L(x)    (x) << 1
#define R(x)    (x) << 1 | 1
#define MID(l, r)   (l + r) >> 1
#define Min(x, y)   (x) < (y) ? (x) : (y)
#define Max(x, y)   (x) < (y) ? (y) : (x)
#define E(x)        (1 << (x))
#define iabs(x)     (x) < 0 ? -(x) : (x)
#define OUT(x)  printf("%I64d\n", x)
#define lowbit(x)   (x)&(-x)
#define Read()  freopen("din.txt", "r", stdin)
#define Write() freopen("dout.txt", "w", stdout);

#define M 30007
#define N 10007

using namespace std;

const int inf = 100000007;
const int mod = 1000000007;

int f
;
int no
,tim
;
int n,m;

void init()
{
for (int i = 0; i <= n; ++i)
{
f[i] = i;
no[i] = 1;
tim[i] = 0;
}
}
int find(int x)
{
if (f[x] != x)
{
int fa = find(f[x]);
tim[x] += tim[f[x]];
f[x] = fa;
}
return f[x];
}
void Union(int x,int y)
{
x = find(x);
y = find(y);
if (x != y)
{
f[x] = y;
no[y] += no[x];
tim[x]++;
}
}
int main()
{
//    Read();
int T;
char op[2];
int cas = 1;
scanf("%d",&T);
while (T--)
{
printf("Case %d:\n",cas++);

scanf("%d%d",&n,&m);
init();
while (m--)
{
scanf("%s",op);
int x,y;
if (op[0] == 'T')
{
scanf("%d%d",&x,&y);
Union(x,y);
}
else
{
scanf("%d",&x);
int rt = find(x);
printf("%d %d %d\n",rt,no[rt],tim[x]);
}
}
}
return 0;
}