PTA小辈分--dfs

#include<bits/stdc++.h>
using namespace std;
//#define int long long
int n,k;
vector<int> a[100001];
int cc[100001];
int bo[100001];
int c;
int d;
int vv[100001];
void dfs(int x,int s)
{
    if(!a[x].size())
    {
        cc[c++]=s;
    }
    else
    {
        for(int i=0;i<a[x].size();i++)
        {
            if(bo[a[x][i]]==0)
            {
                bo[a[x][i]]=1;
                dfs(a[x][i],s+1);
            }
        }
    }
}
void df(int x,int s,int g)
{
    if(!a[x].size())
    {
        if(s==g)
            vv[d++]=x;
    }
    else
    {
        for(int i=0;i<a[x].size();i++)
        {
            if(bo[a[x][i]]==0)
            {
                bo[a[x][i]]=1;
                df(a[x][i],s+1,g);
                //bo[a[x][i]]=0;
            }
        }
    }
}
int main()
{
   cin>>n;
   for(int i=1;i<=n;i++)
   {   int x;
       cin>>x;
       if(x!=-1)
       a[x].push_back(i);
       else k=i;
   }
   bo[k]=1;
   dfs(k,0);
    sort(cc,cc+c);
    cout<<cc[c-1]+1<<endl;
    memset(bo,0,sizeof(bo));
    bo[k]=1;
    df(k,0,cc[c-1]);
    //sort(vv,vv+d);
    //cout<<vv[0];
    //cout<<d;

    for(int i=0;i<d;i++)
    {
        if(i==0) cout<<vv[i];
        else cout<<" "<<vv[i];
    }
//    while(b.size())
//    {
//        int x=b.front();
//        if(a[x].size())
//        {
//            for(int i=0; i<a[x].size(); i++)
//            {
//                b.push(a[x][i]);
//            }
//
//        }
//        else
//        {
//            c++;
//        }
//        b.pop();
//
//    }

}
/*泛舟沧海,立马昆仑*/
/*若你为了错过太阳而哭泣,那么你也将错过群星了*/
/*红了樱桃,绿了芭蕉*/
/*泛舟沧海,立马昆仑*/
/*若你为了错过太阳而哭泣,那么你也将错过群星了*/
/*红了樱桃,绿了芭蕉*/
/*
3 3
0 3
1 2
0 2
*/

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值