题目
本题给定一个庞大家族的家谱,要请你给出最小一辈的名单。
输入格式:
输入在第一行给出家族人口总数 N(不超过 100 000 的正整数) —— 简单起见,我们把家族成员从 1 到 N 编号。随后第二行给出 N 个编号,其中第 i 个编号对应第 i 位成员的父/母。家谱中辈分最高的老祖宗对应的父/母编号为 -1。一行中的数字间以空格分隔。
输出格式:
首先输出最小的辈分(老祖宗的辈分为 1,以下逐级递增)。然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔,行首尾不得有多余空格。
输入样例:
9
2 6 5 5 -1 5 6 4 7
输出样例:
4
1 9
思路
用vector 作二维邻接链表
用dfs广度优先遍历求深度(辈分),depth[100010],代表每个人的深度
根据深度ans输出成员
代码
代码如下(示例):
#include <iostream>
#include <vector>
#include <queue>
#define INF 100010
using namespace std;
vector<int> v[INF];
int depth[INF],n;
// 广度优先遍历
int dfs()
{
int x;
queue<int> q;
// -1下的最高老祖宗只有一个,直接将老祖宗添加入列表
q.push(v[0][0]);
// 当链表空的时候结束
while(!q.empty())
{
x = q.front();
q.pop();
for(int i=0;i<v[x].size();i++)
{
// x下的孩子深度都要加一
depth[v[x][i]] = depth[x]+1;
// 广度优先,将这一层全部加入队列
q.push(v[x][i]);
}
}
// 因为是深度优先遍历,结束的时候一定是最深一层的x,这个时候返回深度即可
return depth[x];
}
int main()
{
scanf("%d",&n);
int m;
// 这里注意编号 是 1-N
for(int i=1;i<=n;i++)
{
scanf("%d",&m);
if(m==-1)
{
m=0; // 在 v的0个存储
// 家谱中辈分最高的老祖宗对应的父/母编号为 -1
// 因此为-1的时候,i就是最高的老祖宗
depth[i]=1;
}
// m 是 i的父母,m链表下的值都是m的孩子
v[m].push_back(i);
}
int ans;
ans = dfs();
cout<<ans<<endl;
int flag = 1;
// 从1开始,不可能为0
for(int i=1;i<=n;i++)
{
if(depth[i]==ans)
{
if(flag==1)
{
printf("%d",i);
flag=0;
}
else
printf(" %d",i);
}
}
return 0;
}