分析:
开始以为是类似并查集的从下向上搜索,但是这样会重复搜索很多数据,时间超时。
所以正确解法是从上到下深度搜索,vector容器的引人非常关键,用空间资源简化了时间资源。
代码:
#include <bits/stdc++.h>
using namespace std;
int generation[100010], ma;
vector <int> v[100010];
void dfs(int a, int depth)
{
ma = max(ma, depth);//求出最小代
generation[a] = depth;
for (int i = 0; i<v[a].size(); i++)
dfs(v[a][i], depth + 1);
}
int main()
{
int n, ancestor;
cin >> n;
for (int i = 1; i <= n; i++)
{
int a;
cin >> a;
if (a == -1)ancestor = i;
else v[a].push_back(i);
}
//输入数据,并把每一个人的孩子存进各自的vector
dfs(ancestor, 1);
//自祖宗向孩子深搜
cout << ma << endl;//开始输出
bool flag = 0;//保证第一个数据前无空格
for (int i = 1; i <= n; i++){
if (generation[i] == ma){
if (flag)cout << ' ' << i;
else{
cout << i;
flag = 1;
}
}
}
return 0;
}