結果

問題 No.994 ばらばらコイン
コンテスト
ユーザー oooooba
提出日時 2021-05-15 20:00:58
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
WA  
実行時間 -
コード長 1,312 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,375 ms
コンパイル使用メモリ 123,620 KB
最終ジャッジ日時 2025-01-21 13:01:37
ジャッジサーバーID
（参考情報） 		judge5 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 11 WA * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <algorithm>
#include <array>
#include <cmath>
#include <cstdio>
#include <deque>
#include <iomanip>
#include <iostream>
#include <numeric>
#include <optional>
#include <queue>
#include <set>
#include <stack>
#include <unordered_map>
#include <unordered_set>
#include <vector>

using namespace std;

int main() {
    int32_t n, k;
    cin >> n >> k;
    vector<vector<int32_t>> tr(n);
    for (auto i = 0; i < n - 1; ++i) {
        int32_t a, b;
        cin >> a >> b;
        --a;
        --b;
        tr[a].push_back(b);
        tr[b].push_back(a);
    }
    if (k > n) {
        cout << -1 << endl;
        return 0;
    }
    int64_t ans = 0;
    queue<pair<int32_t, int32_t>> que;
    que.push({0, 0});
    vector<bool> visited(n);
    visited[0] = true;
    --k;
    while (k > 0) {
        auto p = que.front();
        que.pop();
        int32_t cnt = 0;
        for (auto v : tr[p.first]) {
            if (visited[v])
                continue;
            ++cnt;
            que.push({v, p.second + 1});
            visited[v] = true;
            --k;
            if (k == 0)
                break;
        }
        if (cnt == 0)
            ans += p.second;
    }
    while (!que.empty()) {
        ans += que.front().second;
        que.pop();
    }
    cout << ans << endl;
    return 0;
}
0