Problem Description
有一个研究团队,团队分成许多研究小组,每个小组的一部分成员可能再分成小组。小组的成员只知道自己的组长是谁,而在同一个组长领导下的成员之间却相互不认识。现在这个团队希望有一个程序能统计一下各组长带领小组的规模,即对每一个成员想知道自己及自己带领下的小组有多少人。
Input
测试2行,第1行有1个数字N(0 < N < 2×105),代表小组的人数第2行有N个数a1,a2,…,ai,…,aN,表示第i个人的领导是ai。团队的领导用0表示,说明没有人做他的组长。数据保证没有环路。单独的一个成员视为1个人的小组。
Output
1行,N个数字,表示第i名成员的团队的规模。
Sample Input
6
0 1 2 1 2 2
Sample Output
6 4 1 1 1 1
解题思路
- 算法笔记p328并查集。
- 初始化时所有组员的组长为其自身,录入数据时,对数据0进行单独处理,将0改为自身。
- 遍历计算所有成员的组员数量时,应先记录该成员的组长(记录老组长),将该成员组长改为自己(出队),统计完所有以该成员为组长的组员数后(报数),再将该成员组长改为原先的组长(归队)。
经验总结
- 由于需要计算每个小组长的成员,所以不应该做路径压缩。
- 仅需并查集的查找操作,无需合并操作。
代码实现(C)
#include <stdio.h>
#define MaxSize 200002
// 并查集
int father[MaxSize];
// 初始化每个结点的父亲结点为其自身
void init(int N) {
for (int i = 1; i <= N; ++i)
father[i] = i;
}
// 查找父亲结点(不做路径压缩)
int findFather(int x) {
if (x == father[x])
return x;
else
return findFather(father[x]);
}
// 录入数据
void inputData(int N) {
for (int i = 1; i <= N; ++i) {
scanf("%d", &father[i]);
if (father[i] == 0)
father[i] = i;// 若读入数据为0,则令该结点的父亲结点为其自身(与初始时相同)
}
}
// 计算小组成员个数
void countNumber(int N) {
// 遍历所有成员
for (int i = 1; i <= N; ++i) {
int count = 0;
int faI = father[i]; // 记录成员i的父亲结点
father[i] = i; // 令成员i的父亲结点为其自身,即脱离上级,只计算自己组
for (int j = 1; j <= N; ++j) // 遍历所有成员
if (findFather(j) == i) // 判断成员的父亲结点是否为i
count++; // 记录以i为组长的成员数
father[i] = faI; // 遍历结束后还原i的父亲结点
if (i < N)
printf("%d ", count);
else
printf("%d\n", count);
}
}
int main() {
int N;
while (~scanf("%d", &N)) {
init(N);
inputData(N);
countNumber(N);
}
return 0;
}