結果
問題 | No.994 ばらばらコイン |
ユーザー | Shibuyap |
提出日時 | 2020-02-21 21:31:13 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 54 ms / 2,000 ms |
コード長 | 1,785 bytes |
コンパイル時間 | 1,811 ms |
コンパイル使用メモリ | 172,704 KB |
実行使用メモリ | 19,840 KB |
最終ジャッジ日時 | 2024-10-08 21:13:33 |
合計ジャッジ時間 | 3,336 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 23 |
ソースコード
#include <bits/stdc++.h> #define rep(i,n) for(int i = 0; i < (n); ++i) #define rrep(i,n) for(int i = 1; i <= (n); ++i) #define drep(i,n) for(int i = (n)-1; i >= 0; --i) #define srep(i,s,t) for (int i = s; i < t; ++i) #define rng(a) a.begin(),a.end() using namespace std; typedef long long int ll; typedef pair<int,int> P; typedef vector<int> vi; typedef vector<vi> vvi; typedef vector<ll> vl; typedef vector<P> vp; #define dame { puts("-1"); return 0;} #define yn {puts("Yes");}else{puts("No");} #define MAX_N 200005 vector<int> pre_G[MAX_N]; vector<int> G[MAX_N]; int par_[MAX_N]; int flag_netsukigi[MAX_N]; void make_netsukigi(int r){ queue<int> que; que.push(r); flag_netsukigi[r] = 1; par_[r] = -1; while(que.size() > 0){ int x = que.front(); que.pop(); flag_netsukigi[x] = 1; rep(i,pre_G[x].size()){ if(flag_netsukigi[pre_G[x][i]] == 0){ par_[pre_G[x][i]] = x; G[x].push_back(pre_G[x][i]); que.push(pre_G[x][i]); } } } } int main() { int n , k; cin >> n >> k; rep(i,n-1){ int a, b; scanf("%d%d", &a, &b); a--; b--; // 0-indexにするため. pre_G[a].push_back(b); pre_G[b].push_back(a); } make_netsukigi(0); if(n < k){ cout << -1<< endl; return 0; } cout << k - 1 << endl; return 0; int z[n] = {}; queue<int> que; que.push(0); while(que.size() > 0){ int x = que.front(); que.pop(); rep(i,G[x].size()){ z[G[x][i]] = z[i] + 1; que.push(G[x][i]); } } sort(z,z+n); ll ans = 0; rep(i,k){ ans += z[i]; } cout << ans << endl; return 0; }