結果

問題 No.994 ばらばらコイン
ユーザー zeronosu77108zeronosu77108
提出日時 2020-02-22 18:47:39
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,273 bytes
コンパイル時間 1,207 ms
コンパイル使用メモリ 143,068 KB
実行使用メモリ 9,088 KB
最終ジャッジ日時 2024-05-07 20:31:37
合計ジャッジ時間 2,505 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 AC 31 ms
8,576 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 27 ms
7,168 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
6,940 KB
testcase_17 AC 1 ms
6,944 KB
testcase_18 AC 2 ms
6,940 KB
testcase_19 AC 2 ms
6,944 KB
testcase_20 AC 2 ms
6,944 KB
testcase_21 WA -
testcase_22 WA -
権限があれば一括ダウンロードができます

ソースコード

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