题解 | #快乐节点#

快乐节点

https://www.nowcoder.com/practice/f28bccea385e4cd0a472bfc4859b2290

#include <bits/stdc++.h>


// 多源头广度优先遍历!
using namespace std;

int main() {
    int n, m, u, v;
    cin >> n >> m;

    vector<vector<int>> tree(n + 1, vector<int>());

    vector<bool> vis(n + 1, false);

    for (int i = 1; i <= n - 1; i++) {
        cin >> u >> v;
        tree[u].push_back(v);
    }

    int k, e;

    cin >> k; // 特殊点的个数!

    queue<int> q;

    for (int i = 1; i <= k; i++) {
        cin >> e;
        vis[e] = true;
        q.push(e);
    }

    multiset<int> ret;

    while (!q.empty()) {
        int size = q.size();
        if (m == -1) break;
        while (size--) {
            int pop = q.front();
            q.pop();
            ret.insert(pop);
            for (auto next : tree[pop]) {
                if (!vis[next]) {
                    q.push(next);
                    vis[next] = true;
                }
            }
        }
        m--;
    }
    cout << ret.size() << endl;
    for (auto val : ret) cout << val << " ";
    return 0;
}

全部评论

相关推荐

头像
10-22 20:13
中南大学 Java
序言大家好呀。我是希晨er,一个初入职场的程序猿小登最近上班摸鱼刷到了一篇文章:10年深漂,放弃高薪,回长沙一年有感,还有聊聊30岁大龄程序员过往的心路历程,突然就有点感慨。我如今也做出了和大明哥一样的抉择,只是更早。此外我22年的人生,好像从来没好好记录过。正好现在工作不太忙,就想把这些经历写下来,也希望能得到社区里各位前辈的指点个人背景我是03年出生的西安娃,父母都是普通打工人。刚从中南大学软件工程专业毕业半年,现在在老家的央企过着躺平摆烂的日子成长轨迹从农村到城市的童年我家并不是西安的,只是爸妈在西安上班,从小学之后就把我接到了西安。后来老家房子拆了,爷爷奶奶也搬了过来。农村的生活,我觉...
Yki_:看哭了,恋爱那一段你女朋友说你不够关心她,可你毕竟也愿意遇到矛盾写几千字来和她慢慢分析;说不愿意给她花钱,我感觉可能只是消费观不一样;如果她想留在长沙,也应该提前跟你说开。不过她也许会心疼你放弃大厂offer转向数字马力?我也因为同样的原因有过一段幸福而充满遗憾的感情,不过跟爱情相比确实前途更重要一点。至于offer的选择,换我我也会这么选。把这些旧事记录下来以后,接下来就好好向前看吧,加油兄弟
🍊晨光随笔
点赞 评论 收藏
分享
迷茫的大四🐶:哇靠,哥们,啥认证啊,副总裁实习,这么有实力嘛
一起聊美团
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务