結果

問題 No.994 ばらばらコイン
ユーザー horoyoisawahoroyoisawa
提出日時 2020-02-21 22:33:32
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 813 bytes
コンパイル時間 1,838 ms
コンパイル使用メモリ 171,016 KB
実行使用メモリ 9,856 KB
最終ジャッジ日時 2024-04-17 07:44:09
合計ジャッジ時間 3,228 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 AC 1 ms
5,376 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 2 ms
5,376 KB
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 1 ms
5,376 KB
testcase_21 WA -
testcase_22 WA -
権限があれば一括ダウンロードができます

ソースコード

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