L2-026 小字辈(dfs)

#include<cstdio>
#include<algorithm>
#define foru(i,a,b) for(int i=a;i<=b;i++)
#define sd(a) scanf("%d",&a)
using namespace std;
typedef long long ll;
const int N=1e5+5,M=N*100,INF=0x3f3f3f3f;
struct Edge{int to,nex,k;}edge[M];
int head[N],fin[N],maxk,tot;
void add(int from,int to)
{
    edge[++tot]=(Edge){to,head[from],0};head[from]=tot;
}
void dfs(int x,int k)
{
    maxk=max(maxk,k);
    for(int i=head[x];i;i=edge[i].nex)
    {
        int y=edge[i].to;
        edge[i].k=k+1;
        dfs(y,k+1);
    }
}
int main()
{
    maxk=1;
    int n,dex;scanf("%d",&n);
    foru(i,1,n)
    {
        int x;sd(x);
        if(x==-1)
            dex=i;
        else
            add(x,i);
    }
    if(n==1)//这里需要注意是一个测试点
    {
        printf("1\n1\n");
        return 0;
    }
    dfs(dex,1);
    int cnt=0;
    for(int i=1;i<=tot;i++)
    {
        if(edge[i].k==maxk)
            fin[++cnt]=edge[i].to;
    }
    sort(fin+1,fin+cnt+1);
    printf("%d\n",maxk);
    for(int i=1;i<=cnt;i++)
        printf("%d%c",fin[i],(i==cnt)?'\n':' ');
}