PAT-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

本题可以抽象成一颗树，我们可以将某一个人的孩子存入vector中，因为孩子之间是没有先后顺序的，所以存储了数据之后，我们就可以利用队列实现层次遍历，同时将层次遍历的结果存入结果集中，同时更新最大的Level值，最后遍历结果集，如果结果集中有和level同样值的节点，就将其加入到结果中输出即可，代码如下：

//@author:hairu,wu
//@from:ahut
#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>

using namespace std;
const int max_n=1e5+100;

int Begin;	//老祖宗 
vector<int> sons[max_n];	//存储直接的孩子 
int n;

struct person{
	int id;
	int level;
};

int main(){
	cin >> n;
	for(int i=1;i<=n;i++){
		int parent;
		cin >> parent;
		if(parent==-1){
			Begin=i;
		}else{
			sons[parent].push_back(i);//存储直接孩子 
		}
	}
	int level=0;//最低的辈分 
	vector<int> v;//最后同样level的人结果 
	queue<person> ans;	//存储所有人的层次和ID 
	//将第一个人入队 
	queue<person> q;
	person p;
	p.id=Begin;
	p.level=1;
	q.push(p);
	ans.push(p);
	while(!q.empty()){
		person now=q.front();
		q.pop();
		if(now.level>level){
			level=now.level;
		}
		for(int j=0;j<sons[now.id].size();j++){
			int nextId=sons[now.id][j];
			p.id=nextId;
			p.level=now.level+1;
			q.push(p);
			ans.push(p);
		}
	}
	
	//将同样层次的ID存入结果集 
	cout<<level<<endl;
	while(!ans.empty()){
		person p=ans.front();
		ans.pop();
		if(p.level==level){
//			s.insert(p.id);
			v.push_back(p.id);
		}
	}
	
	sort(v.begin(),v.end());
	for(int i=0;i<v.size();i++){
		cout<<v[i];
		if(i<v.size()-1) cout<<" ";
		else cout<<endl;
	}
	return 0;
}