L2-026 小字辈

L2-026 小字辈 （25 分）

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

输入格式：

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

输出格式：

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

输入样例：

9
2 6 5 5 -1 5 6 4 7

输出样例：

4
1 9

 直接这样暴力深搜会超时，只有16分。

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<string>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<vector>
using namespace std;
#define inf 0x3f3f3f3f
#define LL long long
int a[100005];
struct sb
{
	int i,bf;
}b[100005];
int dfs(int x,int bf)
{
	if(x==-1) return bf;
	return dfs(a[x],bf+1);
}
bool cmp(sb u,sb w)
{
	if(u.bf!=w.bf)
	return u.bf>w.bf;
	else
	return u.i<w.i;
}
int main()
{
	int n,i,j;
	scanf("%d",&n);
	for(i=1;i<=n;i++)	
	scanf("%d",&a[i]);
	for(i=1;i<=n;i++)
	{
		b[i].i=i;
		b[i].bf=dfs(a[i],1);
		//printf("%d %d\n",i,b[i]);
	}
	sort(b+1,b+1+n,cmp);
	printf("%d\n",b[1].bf);
	printf("%d",b[1].i);
	for(i=2;i<=n;i++)
	{
		if(b[i].bf==b[1].bf)
		printf(" %d",b[i].i);
		else break;
	}
	printf("\n");
}

优化一下， 用vector存一辈人。即在递归过程中，如果发现更小的一辈，那么就将v清空，并将这个人的编号添加到v中，如果发现同样是最小一辈的人，则将该人编号添加到v中。

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<string>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<vector>
using namespace std;
#define inf 0x3f3f3f3f
#define LL long long
vector<int> d[100005],v;
int mbf=0;// max 辈分 
void dfs(int x,int bf)
{
	if(bf==mbf) v.push_back(x);
	if(bf>mbf)
	{
		mbf=bf;
		v.clear();
		v.push_back(x);
	}
	for(int i=0;i<d[x].size();i++)
	dfs(d[x][i],bf+1);
}
int main()
{
	int n,i,j,a,root;
	scanf("%d",&n);
	for(i=1;i<=n;i++)	
	{
		scanf("%d",&a);
		if(a==-1)
		root=i;
		else
		d[a].push_back(i);
	}
	dfs(root,1);
	sort(v.begin(),v.end());
	printf("%d\n",mbf);
	for(i=0;i<v.size();i++)
	{
		if(i)printf(" ");
		printf("%d",v[i]);
	}
	printf("\n");
}