小字辈(25 分)

小字辈

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

输入格式：

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

输出格式：

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

输入样例：

9
2 6 5 5 -1 5 6 4 7

输出样例：

4
1 9

思路

• 依次计算家族成员辈分，并用数组记录
• 查找数组中最小辈分数值
• 输出数值和位置
• 2020-9.28更新c++解法

遇到的问题

1.人口总数N（不超过 100 000 的正整数） 数组过大会段错误¹。

解决方法：需要用动态数组。

scanf("%d",&n);
int *pre=(int *)calloc(n,sizeof(int));

2.用递归计算成员辈分超时。
解决方法：第一次计算成员辈分时用数组记录下辈分，下次查找到此成员时加上辈分即可，不用往后查找计算。

2019-11-14，C代码

#include <stdio.h>
#include <stdlib.h>
int count(int i,int *a,int *pre)//递归计算成员辈分
{
	if(a[i]==0)//当第一次计算此成员辈分时
	{
		if(pre[i]!=-1)	return a[i]=count(pre[i],a,pre)+1;//若此成员父辈不是老祖宗，继续找下去，当有返回值时用数组记录此成员辈分
		else return a[i]=1;//找到老祖宗，记录老祖宗辈分，并返回老祖宗辈分“1”
	}
	else return a[i];//第二次查找此成员辈分，直接返回辈分值
}
int main()
{
	int n,i,c,k=0,max=1;
	scanf("%d",&n);
	int *pre=(int *)calloc(n,sizeof(int));//记录父辈位置
	int *a=(int *)calloc(n,sizeof(int));//记录此成员辈分
	for(i=1;i<=n;i++) scanf("%d",&pre[i]);
	for(i=1;i<=n;i++)	//找最小辈分值（数值最大）
	{	
		c=count(i,a,pre);//用递归计算成员i辈分
		if(max<c) max=c;
	}
	printf("%d\n",max);
	for(i=1;i<=n;i++)//按递增顺序输出辈分最小成员编号
	{
		if(a[i]==max)
		{
			if(k!=0) printf(" ");
			printf("%d",i);
			k=1;
		}
	}
	return 0;
}

2020-9.28C++代码

上面的c代码写的我都不想看了，就当个黑历史算了，还是写个c++吧

#include <iostream>
#include <algorithm>
#include <vector> 
using namespace std;
vector<int> p,Rank;//用vector按需求创建数组，也可以直接用int数组开到100005
int rank_find(int x)//找此成员祖先并且用Rank数组记录辈分
{
	if(p[x]==-1) return Rank[x]=1;//是老祖宗，记录第1代并返回1代的辈分
	if(Rank[x]==0) return Rank[x]=rank_find(p[x])+1;//若Rank里没有记录，则递归查找
	else return Rank[x];//Rank里有记录则直接返回
}
int main()
{
	int n,x,Max=-1;
	p.push_back(0);//为了让第i个里的之是i的父亲，这里加个0占个位置
	Rank.push_back(0);
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>x;
		p.push_back(x);
		Rank.push_back(0);
	 }
	for(int i=1;i<=n;i++)
		Max=max(Max,rank_find(i));//找最大辈分
	cout<<Max<<endl;
	for(int i=1,ok=0;i<=n;i++) 
		if(Rank[i]==Max) {
		if(ok==1) cout<<" ";
		ok=1;
		cout<<i;
	}
	return 0;
}

#include <iostream>
#include <algorithm>
#include <vector> 
using namespace std;
int p[100005],Rank[100005];//直接用int数组开到100005
int rank_find(int x)//找此成员祖先并且用Rank数组记录辈分
{
	if(p[x]==-1) return Rank[x]=1;//是老祖宗，记录第1代并返回1代的辈分
	if(Rank[x]==0) return Rank[x]=rank_find(p[x])+1;//若Rank里没有记录，则递归查找
	else return Rank[x];//Rank里有记录则直接返回
}
int main()
{
	int n,x,Max=-1;
	cin>>n;
	for(int i=1;i<=n;i++) cin>>p[i];
	for(int i=1;i<=n;i++)
		Max=max(Max,rank_find(i));//找最大辈分
	cout<<Max<<endl;
	for(int i=1,ok=0;i<=n;i++) 
		if(Rank[i]==Max) {
		if(ok==1) cout<<" ";
		ok=1;
		cout<<i;
	}
	return 0;
}

---

1. 段错误：程序发生段错误，可能是数组越界，堆栈溢出（比如，递归调用层数太多）等情况引起。 ↩︎