小字辈（bfs+一维数组记录深度+set记录当前层的元素）

7-81 小字辈

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

输入格式：

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

输出格式：

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

输入样例：

9
2 6 5 5 -1 5 6 4 7

输出样例：

4
1 9

 思路：这道题是一个家谱，可以抽象成一个树形结构，用vector<int>v[maxn]存储编号为x的的所有孩子编号，v[x][i]表示编号为x的成员的第i个孩子的编号，用bfs依次遍历每一层的元素同时用depth[i]记录编号为i的元素的深度（即：辈分），用set保存当前这一层的所有成员，这样set中最后就保存了最底层（即：最小辈分）的所有成员编号，因为set可以自动去重排序，所以最后输出set中任意一位成员编号i的对应的深度depth[i]，然后依次输出set中的所有成员即可。

完整代码：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e5+5;
vector<int>v[maxn];
set<int>st;
queue<int>q;
int depth[maxn],maxlen=1;

void bfs()
{
    while(!q.empty()){
        int p=q.front();
        q.pop();//bfs基本操作，每次取出队头元素
        for(int i=0;i<v[p].size();i++){//遍历p的所有孩子
            int x=v[p][i];//x为编号为p人的第i个孩子的编号
            q.push(x);//bfs基本操作，每次遍历完一个孩子就将其push进queue
            depth[x]=depth[p]+1;//更新depth[x]的值
            if(maxlen<depth[x]){//每次maxlen更新时，就是刚进入新的一层的时候（因为同一层的深度是一样的,maxlen不会更新）
                maxlen=depth[x];//maxlen记录当前最大深度
                st.clear();//这时候顺便清空st，让st记录新一层的每个成员（st只保存一层的成员）
            }
            st.insert(x);//每次把当前成员塞进set即可
        }
    }
}

signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,root;
    cin>>n;
    for(int i=1;i<=n;i++){
        int x;
        cin>>x;
        if(x==-1){
            root=i;//root标记最顶层，即：老祖宗的编号
            continue;
        }
        v[x].push_back(i);//把x的孩子编号i放入v[x]储存起来
    }
    q.push(root);//bfs从root开始
    depth[root]=1;//root的辈分为1
    st.insert(root);//一开始顶层元素为root
    bfs();
    set<int>::iterator it;
    for(it=st.begin();it!=st.end();it++){
        if(it!=st.begin()){
            cout<<" ";
        }
        else{
            cout<<depth[*it]<<endl;
        }
        cout<<*it;
    }
    cout<<endl;
    return 0;
}