XMU 数据结构实验 小字辈

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

输入格式：

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

输出格式：

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

输入样例：

9
2 6 5 5 -1 5 6 4 7

输出样例：

4
1 9

这道题我自己的思路：

先找到家族里所有没孩子的人（叶子节点），他们中必定包括辈分最小的，同时还有一部分辈分较大但是没有孩子的。

每个叶子节点，一次次返回返回，到老祖宗为止，找到辈分最小，也就是返回次数最多的。

但是，但是，但是

6个数据，最后一个数据超时咯，GG。

先贴我的超时代码

#include<cstdio>
#include<iostream>
#include<queue>
using namespace std;
int N[100005],tmp[100005],candidates[10005],maxx,sum=1;
queue<int> q;                      //存放辈分最小的人
int main(){
	int n,can=0,t=0,index;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>N[i];
		tmp[N[i]]=1;               //以输入数据N[i]为序号的人必定有孩子，为i
	}
	for(int i=1;i<=n;i++)
	{
		if(!tmp[i])
		{
			candidates[++t]=i;	//candidates中即为所有没孩子的人
		}
	}
	for(int i=1;i<=t;i++)
	{
		index=candidates[i];
		while(N[index]!=-1)         //一层层向上，一直到祖宗
		{
			sum++;
			index=N[index];
		}
		if(sum>maxx){
			while(!q.empty())
				q.pop();
			q.push(candidates[i]);
			maxx=sum;
			sum=1;
		}
		else if(sum==maxx){
			q.push(candidates[i]);
			sum=1;
		}
		else sum=1;
	}
	cout<<maxx<<endl;
	cout<<q.front();
	q.pop();
	while(!q.empty())
	{
		cout<<" "<<q.front();
		q.pop();
	}
	return 0;
}

于是乎就又去网上找大佬咯

(21条消息) PAT 7-7 小字辈 （25 分）_【木木】的博客-优快云博客

让我自己再想想…………