結果

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

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define rep(i, n) for(int(i)=0;(i)<(n);(i)++)

int main() {
  int n, k;
  cin >> n >> k;
  if(n < k) cout << -1 << endl;
  else {
    vector<vector<int>> G(n);
    rep(i, n-1) {
      int a, b;
      cin >> a >> b;
      a--;b--;
      G[a].emplace_back(b);
      G[b].emplace_back(a);
    }

    queue<int> que;
    long long total = 0LL;
    vector<int> used(n);
    vector<int> depth(n);
    que.emplace(0);
    int num = 0;
    while(!que.empty()) {
      int v = que.front();que.pop();
      used[v] = 1;
      total += (long long)depth[v];
      num++;
      if(num == k) break;
      for(auto next_v : G[v]) {
        if(used[next_v]) continue;
        depth[next_v] = depth[v] + 1;
        que.emplace(next_v);
      }
    }

    cout << total << endl;
  }
}
0