結果

問題 No.994 ばらばらコイン
コンテスト
ユーザー あるふぁ
提出日時 2020-02-21 21:39:31
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,308 bytes
コンパイル時間 1,697 ms
コンパイル使用メモリ 177,168 KB
実行使用メモリ 9,216 KB
最終ジャッジ日時 2024-10-08 21:24:46
合計ジャッジ時間 3,213 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 7 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using lint = long long;
using pint = pair<int, int>;
using vi = vector<int>;
using vvi = vector<vector<int>>;
#define rep(i, n) for(int i = 0; i < (int)(n); i++)
#define all(v) v.begin(), v.end()

const int MOD = 1000000007;
const int INF = 1 << 30;

int main() {
    int N, K;
    cin >> N >> K;
    vvi G(N+1);
    rep(i, N-1) {
        int a, b;
        cin >> a >> b;
        G[a].push_back(b);
        G[b].push_back(a);
    }

    if (N < K) {
        cout << "-1" << endl;
        exit(0);
    }

    vi dist(N+1, -1);
    queue<int> q;

    dist[1] = 0;
    q.push(1);

    while (!q.empty()) {
        int v = q.front(); // キューから先頭頂点を取り出す
        q.pop();

        // v から辿れる頂点をすべて調べる
        for (int nv : G[v]) {
            if (dist[nv] != -1) continue; // すでに発見済みの頂点は探索しない

            // 新たな白色頂点 nv について距離情報を更新してキューに追加する
            dist[nv] = dist[v] + 1;
            q.push(nv);
        }
    }

    sort(dist.begin()+1, dist.end());

    int res = 0;

    rep(i, K) {
        res += dist[i+1];
    }

    cout << res << endl;
    return 0;
}
0