文章目录
- 题目描述
- 思路
- AC代码
题目描述
思路
深度优先搜索
具体流程
1.读入每个人的祖先,标记辈分最高的老祖宗对应的下标pos
2.从pos开始dfs,每次判断当前遍历的深度,如果>原来的深度,更新,并将存储最小辈分的数组清零;如果=原来深度,则将当前的人加入数组;之后继续搜索该人对应的孩子
AC代码
#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
vector<int> v[N];
vector<int> res;
int cnt = -1;
void dfs(int x, int step)
{
if(step > cnt)
{
cnt = step;
res.clear();
}
if(step == cnt) res.push_back(x);
for(int i = 0; i < v[x].size(); i ++) dfs(v[x][i], step + 1);
return;
}
int main()
{
int n, pos;
cin >> n;
for(int i = 1; i <= n; i ++)
{
int x;
cin >> x;
if(x == -1) pos = i;
else v[x].push_back(i);
}
dfs(pos, 1);
cout << cnt << endl;
for(int i = 0; i < res.size(); i ++)
{
if(i != res.size() - 1) cout << res[i] << " ";
else cout << res[i] << endl;
}
return 0;
}