#432687 (C++14) No.994 ばらばらコイン

• 問題: No.994 ばらばらコイン
• ユーザー: utouto97
• 提出日時: 2020-02-21 21:41:19
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,088 bytes
• コンパイル時間: 1,655 ms
• コンパイル使用メモリ: 174,452 KB
• 実行使用メモリ: 10,240 KB
• 最終ジャッジ日時: 2024-10-08 21:26:40
• 合計ジャッジ時間: 3,288 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	WA	-
testcase_02	AC	67 ms; 8,960 KB
testcase_03	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	AC	50 ms; 8,436 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	3 ms; 6,816 KB
testcase_17	AC	3 ms; 6,820 KB
testcase_18	AC	3 ms; 6,820 KB
testcase_19	AC	3 ms; 6,820 KB
testcase_20	AC	3 ms; 6,820 KB
testcase_21	WA	-
testcase_22	WA	-

ソースコード

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

#define int long long
#define rep(i, l, r) for (int i = (int)(l); i < (int)(r); i++)
#define all(x) (x).begin(), (x).end()
#define sz(x) ((int)x.size())
template <class T> bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <class T> bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return 1;
  }
  return 0;
}

template <class T> using V = vector<T>;
using P = pair<int, int>;

/*
 *  */

int n, k;
V<int> es[101010];
V<int> dist;

void dfs(int v, int p){
  for(int to : es[v]){
    if(to == p) continue;
    chmin(dist[to], dist[v] + 1);
    dfs(to, v);
  }
}

signed main() {
  cin >> n >> k;
  rep(i, 0, n-1){
    int a, b;
    cin >> a >> b;
    a--; b--;
    es[a].emplace_back(b);
    es[b].emplace_back(a);
  }

  if(k > n){
    cout << -1 << endl;
    return 0;
  }

  dist.assign(n, 1LL<<60);
  dist[0] = 0;
  dfs(0, -1);
  sort(all(dist));

  int ans = 0;
  rep(i, 0, k) ans += dist[i];

  cout << ans << endl;

//  rep(i, 0, n) cout << dist[i] << " ";
//  cout << endl;

  return 0;
}