L2-2 小字辈
分数 25
本题给定一个庞大家族的家谱,要请你给出最小一辈的名单。
输入格式:
输入在第一行给出家族人口总数 N(不超过 100 000 的正整数) —— 简单起见,我们把家族成员从 1 到 N 编号。随后第二行给出 N 个编号,其中第 i 个编号对应第 i 位成员的父/母。家谱中辈分最高的老祖宗对应的父/母编号为 -1。一行中的数字间以空格分隔。
输出格式:
首先输出最小的辈分(老祖宗的辈分为 1,以下逐级递增)。然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔,行首尾不得有多余空格。
输入样例:
9
2 6 5 5 -1 5 6 4 7
输出样例:
4
1 9
题解
根据题意存图,然后dfs搜他丫的。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define endl '\n'
int n;
struct node{
int fa;
int son=0;
int deep=1;//记录深度(辈分)
}s[100005];
int v[100005];
vector<int> g[100005];
int minn=0;
int root;
vector<int> ans;
void dfs(int now,int step)//搜他丫的
{
s[now].deep=step;
if(s[now].son==0)//到小辈了就返回
{
minn=max(minn,step);
return;
}
if(v[now]) return;
v[now]=1;
for(int i=0;i<g[now].size();i++)
{
//cout<<g[now][i]<<"*";
dfs(g[now][i],step+1);
}
}
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
cin>>n;
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
s[i].fa=x;
s[x].son=i;
if(x==-1)
{
root=i;//找到根节点
}
else
{
g[x].push_back(i);
}
}
dfs(root,1);
cout<<minn<<endl;
for(int i=1;i<=n;i++)
{
if(s[i].son==0 && s[i].deep==minn)
{
ans.push_back(i);
}
}
int op=0;
for(auto k:ans)
{
op++;
if(op!=ans.size())
{
cout<<k<<" ";
}
else
{
cout<<k<<endl;
}
}
return 0;
}