本题给定一个庞大家族的家谱,要请你给出最小一辈的名单。
输入格式:
输入在第一行给出家族人口总数N
(不超过 100 000 的正整数) —— 简单起见,我们把家族成员从 1 到N
编号。随后第二行给出 N
个编号,其中第i
个编号对应第 i
位成员的父/母。家谱中辈分最高的老祖宗对应的父/母编号为 -1
。一行中的数字间以空格分隔。
输出格式:
首先输出最小的辈分(老祖宗的辈分为 1
,以下逐级递增)。然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔,行首尾不得有多余空格。
输入样例:
9
2 6 5 5 -1 5 6 4 7
输出样例:
4
1 9
参照着柳婼的代码写的,v数组存储的是每个编号的所有孩子,s容器里存储每一层的孩子,所以进行DFS进行到最后,s数组里存储的是最下层,也就是最小辈的所有成员。
DFS方法里:如果level大于了maxl,说明进行到了更下一辈,需要将s清空以存储这一层所有成员;如果level等于maxl,说明还是在同一层,只需要将该结点继续存储在s的这一层。
柳神的代码博客链接
AC代码:
#include<bits/stdc++.h>
using namespace std;
set<int> s;
vector<vector<int>> v;
int maxl = 1;
void dfs(int root,int level){
if(level == maxl){
s.insert(root);
}else if(level > maxl){
maxl = level;
s.clear();
s.insert(root);
}
for(int i = 0;i < v[root].size();i++){
dfs(v[root][i],level+1);
}
}
int main(){
int N;
cin >> N;
v.resize(N+10);
int root;
for(int i = 1;i <= N;i++){
int x;
cin >> x;
if(x == -1){
root = i;
continue;
}
v[x].push_back(i);
}
dfs(root,1);
cout << maxl << endl;
int flag = 0;
for(auto it = s.begin();it != s.end();it++){
if(flag++)
cout << " ";
cout << *it;
}
}