gplt L2-026. 小字辈（bfs）

话不多说 10钟的事 贴代码

#include<bits/stdc++.h> //L2-026. 小字辈（bfs stl）
using namespace std;    //没什么好讲的  就bfs
const int maxn = 1e5 + 5;//虽然队友简单粗暴递归dfs也过了 限时是真的宽容啊
int level[maxn], fat[maxn];//这个fat原来是想辅助标记父亲节点的 后来发现不用了
set<int>ans, son[maxn];//简单粗暴set存儿子节点和答案 stl大法好
int main(){
    int n, fa, i, maxl, t, top, f;
    cin >> n;
    for(i = 1; i <= n; i ++){//读入 全部把对应的子节点扔到对应的set里
        cin >> t;
        if(t == -1)
            fa = i;
        else{
            fat[i] = t;
            son[t].insert(i);
        }
    }
    level[fa] = 1;
    queue<int>q;
    q.push(fa);
    maxl  = 1;
    while(!q.empty()){//bfs一遍 找到最小辈分
        top = q.front();
        q.pop();
        set<int>::iterator it;
        for(it = son[top].begin(); it != son[top].end(); it ++){
            q.push(*it);
            level[*it] = level[top] + 1;
        }
        maxl = max(maxl, level[top]);
        // cout << top << " ";
    }
    //for(i = 1; i <= n ; i ++)
        // cout << level[i] << " ";
    cout << maxl << endl;
    f = 0;
    q.push(fa);//注意这个push
    while(!q.empty()){//再bfs一遍 存答案排序
        top = q.front();
        q.pop();
        set<int>::iterator it;
        for(it = son[top].begin(); it != son[top].end(); it ++){
            q.push(*it);
        }
        if(level[top] == maxl)
            ans.insert(top);
    }
    set<int>::iterator it;//输出
    for(it = ans.begin(); it != ans.end(); it ++){
        if(f) cout << " ";
        else f = 1;
        cout << *it;
    }
    return 0;
}