L2-026 小字辈BFS

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

输入格式：

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

输出格式：

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

输入样例：

9
2 6 5 5 -1 5 6 4 7

输出样例：

4
1 9

从孩子向上找祖先

 

#include<bits/stdc++.h>
using namespace std;
int n;
const int N=1e6+5;
struct node{
	int id;
	int parten;
	int sum;
}a[N];
int vis[N];
int b[N];
bool cmp(node x,node y){
	if(x.sum==y.sum){
		return x.id<y.id;
	}
	return x.sum>y.sum;
}
void bfs(node x){
	queue<node>q;
	q.push(x);
	stack<node>s;
	while(!q.empty()){
		node t;
		t=q.front();
		q.pop();
		s.push(t);
		vis[t.id]=1;
		if(t.parten!=-1&&vis[t.parten]==0){
			q.push(a[t.parten]);
		}
	}
	while(!s.empty()){
		node t=s.top();
		s.pop();
		a[t.id].sum=a[t.parten].sum+1;
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		a[i].id=i;
		cin>>a[i].parten;
		if(a[i].parten==-1){
			a[i].sum=1;
		}
	}
	for(int i=1;i<=n;i++){
		if(vis[i]==0){
			bfs(a[i]);
		}
	}
	sort(a+1,a+1+n,cmp);
    int j=0;
	int cc=a[1].sum;
	cout<<a[1].sum<<endl;
	for(int i=1;i<=n;i++){
		if(a[i].sum==cc){
			b[j++]=a[i].id;
		}
	}
    for(int i=0;i<j;i++){
        if(i==0){
            cout<<b[i];
        }
        else{
            cout<<" "<<b[i];
        }
    }
//	for(int i=1;i<=n;i++){
//		cout<<a[i].sum<<" ";
//	}
	return 0;
}