結果

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

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<cstring>
#include<math.h>
#include<bitset>
#include<queue>
#include<set>
#include<iomanip>
#include<assert.h>
#include<string>
#include<cstdlib>
#include<cctype>
#include<unordered_map>
#include<sstream>
#include<numeric>
#include<stack>
// #include<bits/stdc++.h>
using namespace std;
using ll = long long;
using P = pair<int,int>;
constexpr ll INF = 9e18;
constexpr int inf = 1e9;
const double INFD = 1e100;
const ll mod = 1000000007;
const double PI = 3.1415926535897932384626433832795028841971;
const int dx[4] = {1, 0, -1, 0};
const int dy[4] = {0, 1, 0, -1};
template<class T> inline bool chmin(T& a, T b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
};
template<class T> inline bool chmax(T& a, T b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
};

// intやllをstringに変換
template<typename T> inline string toString(const T &a){
  ostringstream oss;
  oss << a;
  return oss.str();
};
// ----------------------------------------------------------------------------

int main(){
  int N,K;
  cin >> N >> K;
  vector<vector<int>> G(N);
  for(int i=0; i<N-1; i++){
    int a,b;
    cin >> a >> b;
    a--;
    b--;
    G[a].push_back(b);
    G[b].push_back(a);
  }
  if(K > N){
    cout << -1 << endl;
    return 0;
  }
  queue<int> que;
  que.push(0);
  vector<int> dist(N,-1);
  dist[0] = 0;
  ll ans = 0;
  while(que.size()){
    int v = que.front();
    que.pop();
    for(int nv: G[v]){
      if(dist[nv] != -1) continue;
      dist[nv] = dist[v] + 1;
      if(K > 1){
        K--;
        ans += dist[nv];
      }else{
        break;
      }
      que.push(nv);
    }
  }
  cout << ans << endl;
  return 0;
}
0