首先建立一个二维动态数组,存储每个编号的孩子。建立一个一维数组用来记录每个编号的是第几代
根据这个二维数组也能直接找到根结点的编号,然后从根节点开始深搜就可以了。代码如下:
#include<bits/stdc++.h>
using namespace std;
const int maxn=100010;
int n,a[maxn],x,root,ans,flag;
vector<int>v[maxn];
void DFS(int i,int cnt)
{
if(v[i].size()==0) {ans=max(ans,cnt);return;}
for(int j=0;j<v[i].size();j++)
{
int u=v[i][j];
a[u]=cnt+1;
DFS(u,cnt+1);
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>x;
if(x>0) v[x].push_back(i);
else root=i;
}
a[root]=1;
DFS(root,1);
cout<<ans<<endl;
for(int i=1;i<=n;i++)
if(a[i]==ans)
{
if(flag) cout<<" "<<i;
else {cout<<i;flag=1;}
}
}