L2-026 小字辈 (25分)(DFS思路+极短代码)

首先建立一个二维动态数组，存储每个编号的孩子。建立一个一维数组用来记录每个编号的是第几代
根据这个二维数组也能直接找到根结点的编号，然后从根节点开始深搜就可以了。代码如下：

#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;}
		}
}