这是一道考察并查集的路径压缩的题。题意:在悟空的世界,有N个龙珠和N个城市(编号从1到N),神龙最开始把每颗龙珠都放在对应编号的城市。悟空要去收集龙珠,但是这些龙珠有时候是会被转移的。你需要告诉悟空一些有关龙珠的信息才行。现在又T组测试,每组测试都有一个N(龙珠和城市的数量)和Q(操作行为的数量),操作行为有两种:
T A B,将编号为A的龙珠所在城市的所有龙珠转移到编号为B的龙珠所在的城市,两个城市不同
Q A,悟空要知道X(龙珠A所在城市的编号),Y(编号为X的城市里的龙珠数)以及Z(编号为A的龙珠转移的次数)
解题思路:首先根据每个龙珠的编号我们都需要知道3个信息,所在城市的编号,所在城市的龙珠数以及该龙珠被转移过的次数。首先我们分析一下T操作,首先两个城市一定不相同,那么将所有龙珠都移动到另一个城市时,肯定有一个龙珠本来就在那个城市且一次都没有被移动过,这个龙珠就是根节点。那么同理,根节点龙珠肯定从来没被移动过而且不会出现把这个城市的所有龙珠全部移动到空城市这种情况。有了这样的规律,我们可以通过维护根节点所在城市龙珠数来查找这个城市每个龙珠的X和Y。至于Z,路径压缩时我们要把对应节点压缩到根节点之下,对应的Z也要进行同步更新,每次移动时我们只将被移动的城市的根节点的移动次数+1,那么当路径压缩时沿途节点的移动次数之和(当然也包括待压缩节点本身)就是这个节点真正的被移动次数。
接下来是解题代码:
1 #include <stdio.h> 2 #include <stdlib.h> 3 #define N 10001 4 5 int bleg[N]; //存储当前节点的父节点 6 int tran[N]; //存储被移动的次数 7 int city[N]; //存储所在的城市 8 int bnum[N]; //存储在同一地点的龙珠总数 9 int t; 10 int n, m; 11 12 void Init(); //初始化 13 14 int Find(int x); //并查集查找 15 16 void Union(int x, int y); //并查集合并,也就是T操作 17 18 void Q(int x); 19 20 int main() 21 { 22 int tn = 1; 23 int i, a, b; 24 char ch; 25 scanf("%d", &t); 26 while (t--) 27 { 28 scanf("%d %d", &n, &m); 29 Init(); 30 printf("Case %d:\n", tn++); 31 for (i=0; i<m; ++i) 32 { 33 scanf(" %c", &ch); 34 if (ch == 'T') 35 { 36 scanf("%d %d", &a, &b); 37 Union(a, b); 38 } 39 else 40 { 41 scanf("%d", &a); 42 Q(a); 43 } 44 } 45 } 46 return 0; 47 } 48 49 void Init() //初始化 50 { 51 int i; 52 for (i=1; i<=n; ++i) 53 { 54 bleg[i] = city[i] = i; //初始化父节点和所在城市 55 tran[i] = 0; 56 bnum[i] = 1; 57 } 58 return; 59 } 60 61 int Find(int x) //并查集查找 62 { 63 int y = x; 64 int z; 65 int ntran = 0; 66 while (y != bleg[y]) 67 { 68 ntran += tran[y]; //存储沿途节点的被移动次数总和 69 y = bleg[y]; 70 } 71 while (x != bleg[x]) 72 { 73 z = bleg[x]; 74 ntran -= tran[x]; //为下个节点更新移动次数做准备 75 bleg[x] = y; //压缩到根节点之下 76 tran[x] += ntran; //移动次数同步更新 77 x = z; 78 } 79 return y; 80 } 81 82 void Union(int x, int y) //并查集合并,也就是T操作 83 { 84 int fx = Find(x); 85 int fy = Find(y); 86 bleg[fx] = fy; 87 ++tran[fx]; //被移动的根节点的被移动次数加1 88 bnum[fy] += bnum[fx]; //更新城市龙珠数 89 return; 90 } 91 92 void Q(int x) 93 { 94 int fx = Find(x); 95 printf("%d %d %d\n", city[fx], bnum[fx], tran[x]); 96 return; 97 }