結果

問題 No.994 ばらばらコイン
ユーザー zeronosu77108
提出日時 2020-02-22 18:47:39
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,273 bytes
コンパイル時間 1,129 ms
コンパイル使用メモリ 144,340 KB
実行使用メモリ 8,960 KB
最終ジャッジ日時 2024-11-30 16:31:50
合計ジャッジ時間 2,268 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 7 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <iomanip>
#include <vector>
#include <utility>
#include <map>
#include <algorithm>
#include <queue>
#include <cmath>
#include <numeric>
#include <set>

using namespace std;
struct aaa{aaa(){cin.tie(nullptr); ios::sync_with_stdio(false); cout<<fixed<<setprecision(20);};}aaa;
template <class T>ostream &operator<<(ostream &o,const vector<T>&v){o<<"{";for(int i=0;i<(int)v.size();i++)o<<(i>0?", ":"")<<v[i];o<<"}";return o;}
#define debug(v) {cerr<<"\033[1;36m[debug]\033[m "<<#v<<" : "<<v<<endl;}

using int64 = long long;
using P = pair<int,int>;

int bfs(int& n, int& k, const vector<vector<int>>& g) {
    int res = 0;
    vector<bool> visited(n+1,false);
    queue<P> q;
    q.emplace(1,0);

    while(!q.empty() && k > 0) {
        int u,i;
        tie(u,i) = q.front(); q.pop();

        if (visited[u]) continue;
        visited[u] = true;

        k--; res += i;
        for (auto v : g[u]) {
            q.emplace(v, i+1);
        }
    }

    return res;
}

int main() {
    int n,k;
    cin >> n >> k;
    vector<vector<int>> g(n+1);
    for (int i=0; i<n-1; i++) {
        int a,b;
        cin >> a >> b;
        g[a].push_back(b);
        g[b].push_back(a);
    }

    int ans = bfs(n, k, g);

    cout << (k>0? -1 : ans) << endl;
}
0