PTA 小字辈 C语言

本题给定一个庞大家族的家谱，要请你给出最小一辈的名单。

输入格式：
输入在第一行给出家族人口总数 N（不超过 100 000 的正整数） —— 简单起见，我们把家族成员从 1 到 N 编号。随后第二行给出 N 个编号，其中第 i 个编号对应第 i 位成员的父/母。家谱中辈分最高的老祖宗对应的父/母编号为 -1。一行中的数字间以空格分隔。

输出格式：
首先输出最小的辈分（老祖宗的辈分为 1，以下逐级递增）。然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔，行首尾不得有多余空格。

输入样例：

9
2 6 5 5 -1 5 6 4 7

输出样例：

4
1 9

代码实现:

#include<stdio.h>
#define MaxSize 100005
int Father[MaxSize];
int seniority[MaxSize]={-1};
int k=0;
int s[MaxSize];
int FindFather(int x){
	if(Father[x]==-1)//若为大长辈，辈分为1
	seniority[x]=1;
	else{
		seniority[x]=FindFather(Father[x])+1;//若不是大长辈，辈分为其爹+1
	}
	return seniority[x];//返回x的辈分
}
int main() {
	int N, i;
	scanf("%d", &N);
	for (int i = 1; i <= N; i++) {
		scanf("%d", &Father[i]);
	}
	for(int i=1;i<=N;i++){
		FindFather(i);
	}
	int max=-1;
	for(int i=1;i<=N;i++){//找到辈分最小，也就是seniority最大的
		if(seniority[i]>max)
		max=seniority[i];
	}
	printf("%d\n",max);
	for(int i=1;i<=N;i++){
		if(seniority[i]==max)//将小辈存入s数组
		s[k++]=i;
	}
	for(i=0;i<k-1;i++)//输出
	printf("%d ",s[i]);
	printf("%d",s[i]);
}