PTA天梯 L2-026 小字辈 (25分)

题目链接：点击进入

思路

找到根节点，dfs向下搜索，每次更新最大深度，同时记录此时最大深度的节点（vector）

代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
vector<int>v[maxn],ans;
int maxx;
void dfs(int pos,int t)
{
    for(int i=0;i<v[pos].size();i++) 
    {
    	dfs(v[pos][i],t+1);
	}
    if(t>maxx)
    {
        maxx=t;
        ans.clear();
        ans.push_back(pos);
    }
    else if(maxx==t)
    {
    	ans.push_back(pos);
	}
}
int main()
{
    int n,x,pos;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>x;
        if(x==-1) pos=i;
        else v[x].push_back(i);
    }
    dfs(pos,1);
    sort(ans.begin(),ans.end());
    cout<<maxx<<endl;
    for(vector<int>::iterator it=ans.begin();it!=ans.end();it++)
    {
    	if(it==ans.begin()) cout<<*it;
    	else cout<<' '<<*it;
	}
    return 0;
}