2018华东师范大学上机（二）：研究小组数量

任何疑问、意见、建议请留言公众号：一航代码

题目描述：

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

输入格式：

    测试2行，第1行有1个数字N(0<N<2×105)(0<N<2×105)，代表小组的人数第2行有N个数a1,a2……ai……aN，表示第i个人的领导是ai。团队的领导用0表示，说明没有人做他的组长。数据保证没有环路。单独的一个成员视为1个人的小组。

输出格式：

    1行，N个数字，表示第i名成员的团队的规模。

输入样例：

6
0 1 2 1 2 2

输出样例：

6 4 1 1 1 1

解决方法：

（1）代码实现：

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <math.h>
using namespace std;

int main()
{
    int n;
    cin >> n;
    vector<int> count(n + 1, 1); //记录下标为boss，值为小组成员数
    vector<int> a(n + 1, 0);
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
    }
    for (int i = 1; i <= n; i++)
    {
        if (a[i] == 0)
        {
            continue;
        }
        else
        {
            int k = i; //核心代码
            while (a[k] != 0)
            {                 //a[k]寻找k节点的父节点,父节点为0，则父节点是最终BOSS，根节点
                int j = a[k]; //每次向上遍历，记录节点k，的父节点，并将父节点的小组成员数加一
                count[j]++;
                k = j; //将k令成父节点，继续向上遍历
            }
        }
    }
    for (int i = 1; i < n; i++)
    {
        cout << count[i] << " ";
    }
    cout << count[n] << endl;

    system("pause");
    return 0;
}