7-10 小字辈 (25分)

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

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

输入样例：
9
2 6 5 5 -1 5 6 4 7

输出样例：
4
1 9

我先说一下一开始超时的算法吧，用了dp的思想，降低了一些复杂度，但是，没用，依然超时。

#include <bits/stdc++.h>
using namespace std;
int a[100010],b[100010],c[100010];
int n,ma=0;
int f(int i)
{
    if(a[i]==-1)
        b[i]=1;
    else
        b[i]=f(a[i])+1;
    if(b[i]>ma)
        ma=b[i];
    return b[i];
}
int main()
{
    int i,top=1;
    scanf("%d",&n);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        b[i]=0;
    }
    for(i=1;i<=n;i++)
    {
        if(b[i]==0)
        {
            f(i);
        }
    }
    printf("%d\n",ma);
    for(i=1;i<=n;i++)
    {
        if(b[i]==ma)
        {
            c[top++]=i;
        }
    }
    top--;
    for(i=1;i<top;i++)
        printf("%d ",c[i]);
    printf("%d\n",c[top]);
    return 0;
}

后面的代码时抄的，用的bfs，
他在一开始储存数据的方式就和我不一样。
牛呀！
原博客地址

#include <bits/stdc++.h>
using namespace std;

int n,t,mx,f,lzz,bfe[100005];
vector <int> vec[100005];
  /*
     vector   动态数组。
     这是一个二维数组，它的第一维大小是固定的100005，第二维是动态的。
     vec[i].size(),返回这个以为数组的大小，即元素个数
     queue， 队列，queue<int>q;
     q.push(tiem)  把item放进队列
     q.front()     返回队首元素，但不会删除
     q.pop()       删除队首元素
     q.back()      返回队尾元素
     q.size()      返回元素个数
     q.empty()     检查队列是否为空
  */
  /*
      vec，这里用于存储它孩子的编号。
      bef，用于储存他自己的辈分，就是第几辈
  */
void bfs(int id){
	bfe[id] = mx = 1;
	queue <int> Q;
	Q.push(id);
	while(!Q.empty()){
		int fnt = Q.front();
		Q.pop();
		int len=vec[fnt].size();
		for(int i = 0;i < len;i++){
			bfe[vec[fnt][i]] = bfe[fnt] + 1;
			mx = max(mx,bfe[vec[fnt][i]]);
			Q.push(vec[fnt][i]);
		}
	}
	return;
}

int main(){
	scanf("%d",&n);
	for(int i = 1;i <= n;i++){
		scanf("%d",&t);
		if(t == -1){
			lzz = i;
		}
		else{
			vec[t].push_back(i);
			//在尾部添加一个数据。
		}
	}

	bfs(lzz);

	f = 0;
	cout << mx << endl;
	for(int i = 1;i <= n;i++){
		if(bfe[i] == mx){
			if(f == 0){
				printf("%d",i);
				f = 1;
			}
			else{
				printf(" %d",i);
			}
		}
	}
	cout << endl;
	return 0;
}

感谢大佬给的第一段代码修改版！

#include <bits/stdc++.h>
using namespace std;
int a[100010],b[100010],c[100010];
/*
    a[i],储存原始数据,即i个节点的直系祖辈。
    b[i],记录它是第几辈
    c[i],保存最小辈的编号

*/
int n,ma=0;
int f(int i)
{
    if(b[i]!=0)
        return b[i];
    //此处做修改，函数自身调用时，没必要每次都回到调用到祖辈
    if(a[i]==-1)
        b[i]=1;
    else
        b[i]=f(a[i])+1;
    if(b[i]>ma)
        ma=b[i];
    return b[i];
}

int main()
{
    int i,top=1;
    scanf("%d",&n);
    for(i=1;i<=n;i++)
        scanf("%d",&a[i]);

    for(i=1;i<=n;i++)
        if(b[i]==0)
            f(i);

    printf("%d\n",ma);
    for(i=1;i<=n;i++)
    {
        if(b[i]==ma)
        {
            c[top++]=i;
        }
    }
    top--;
    for(i=1;i<top;i++)
        printf("%d ",c[i]);
    printf("%d\n",c[top]);
    return 0;
}