任何疑问、意见、建议请留言公众号:一航代码
题目描述:
有一个研究团队,团队分成许多研究小组,每个小组的一部分成员可能再分成小组。小组的成员只知道自己的组长是谁,而在同一个组长领导下的成员之间却相互不认识。现在这个团队希望有一个程序能统计一下各组长带领小组的规模,即对每一个成员想知道自己及自己带领下的小组有多少人。
输入格式:
测试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;
}