L2-026 小字辈

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

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

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

9
2 6 5 5 -1 5 6 4 7

输出样例：

4
1 9

思路

• 1.维护每个父节点的子节点
  如：样例（[父节点：子节点]）[5:3,4,6] [4:8] [6:2,7] [2:1] [7:9]
• 2.记录最高的祖宗编号，然后按层次遍历，直到没有下一代为止
• 3.对最后一代排序，输出

AC代码

#include<bits/stdc++.h>
using namespace std;
//#define int long long
#define fo(i,a,b) for(int i=a;i<b;i++)
#define lop(i,a,b) for(int i=a;i<=b;i++)
#define QAQ(vec) for(int i=0;i<N;i++){cout<<vec[i]<<" ";}cout<<endl;
#define MX 100007
vector<vector<int>> mp;
int N,start,b,cnt=0;
int main(){
    cin>>N;
    mp=vector<vector<int>>(N+1);
    lop(i,1,N){
        cin>>b;
        if(b==-1){
            start=i;
            continue;
        }
        mp[b].push_back(i);
    }
    //把vector当queue用
    vector<int> q;
    q.push_back(start);
    //最后一代开始的位置
    int pos=0;
    while(true){
        cnt++;//辈分
        int size=q.size();
        for(int i=pos;i<size;i++){
            int fro=q[i];
            //将当前节点的孩子加入队列
            for(auto&it:mp[fro]){
                q.push_back(it);
            }
        }
        //没有下一代
        if(size==q.size())break;
        pos=size;
    }
    sort(q.begin()+pos,q.end());
    cout<<cnt<<endl;
    for(int i=pos;i<q.size();i++){
        if(i>pos)cout<<" ";
        cout<<q[i];
    }

    return 0;
}