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

作者: 陈越

单位: 浙江大学

时间限制: 400 ms

内存限制: 64 MB

代码长度限制: 16 KB

解析  简单bfs

代码：

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
struct People
{ 
    int par;
    int rank;
    vector<int> son;
};
People a[100010];
int  dfs(int x)
{

    queue<People> p;
    p.push(a[x]);
    People q;
    while(!p.empty())
    {
        q=p.front();
        p.pop();
        int i;
        for(i=0;i<q.son.size();i++)
        {   
            a[q.son[i]].rank=q.rank+1;
            p.push(a[q.son[i]]);
        }
    }
    return q.rank;
}
int main()
{
    
   int N;
   cin>>N;
   int i;
   int index;
   for(i=1;i<=N;i++)
   {
       cin>>a[i].par;
       if(a[i].par!=-1)
       {
           a[a[i].par].son.push_back(i);
       }
       else
       {
           index=i;
           a[i].rank=1;
       } 
   }
   int max=dfs(index);
   cout<<max<<endl;
   int it=0;
   for(i=1;i<=N;i++)
   {
       if(a[i].rank==max)
       {
           if(it==0)
           {
               cout<<i;
               it=1;
           }
           else
            cout<<" "<<i;
       }
   }
   cout<<endl;
   return 0;
   
}