2018年华东师范大学研究生复试机试题：研究小组数量

Problem Description

有一个研究团队，团队分成许多研究小组，每个小组的一部分成员可能再分成小组。小组的成员只知道自己的组长是谁，而在同一个组长领导下的成员之间却相互不认识。现在这个团队希望有一个程序能统计一下各组长带领小组的规模，即对每一个成员想知道自己及自己带领下的小组有多少人。

Input

测试2行，第1行有1个数字N(0 < N < 2×10⁵)，代表小组的人数第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

解题思路

1. 算法笔记p328并查集。
2. 初始化时所有组员的组长为其自身，录入数据时，对数据0进行单独处理，将0改为自身。
3. 遍历计算所有成员的组员数量时，应先记录该成员的组长（记录老组长），将该成员组长改为自己（出队），统计完所有以该成员为组长的组员数后（报数），再将该成员组长改为原先的组长（归队）。

经验总结

1. 由于需要计算每个小组长的成员，所以不应该做路径压缩。
2. 仅需并查集的查找操作，无需合并操作。

代码实现（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;
}