結果

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

ソースコード

diff # 

// #define DEBUGGING
#include <bits/stdc++.h>
#define endl '\n'
#define ALL(V) begin(V), end(V)
#define ALLR(V) rbegin(V), rend(V)
using ll = std::int64_t;
using ull = std::uint64_t;
using PLL = std::pair<ll, ll>;
using TLL = std::tuple<ll, ll, ll>;
template <typename T> using V = std::vector<T>;
template <typename T> using VV = V<V<T>>;
template <typename T> const T& var_min(const T &t) { return t; }
template <typename T> const T& var_max(const T &t) { return t; }
template <typename T, typename... Tail> const T& var_min(const T &t, const Tail&... tail) { return std::min(t, var_min(tail...)); }
template <typename T, typename... Tail> const T& var_max(const T &t, const Tail&... tail) { return std::max(t, var_max(tail...)); }
template <typename T, typename... Tail> void chmin(T &t, const Tail&... tail) { t = var_min(t, tail...); }
template <typename T, typename... Tail> void chmax(T &t, const Tail&... tail) { t = var_max(t, tail...); }
template <typename T> const T& clamp(const T &t, const T &low, const T &high) { return std::max(low, std::min(high, t)); }
template <typename T> void chclamp(T &t, const T &low, const T &high) { return t = clamp(t, low, high); }
namespace init__ { struct InitIO { InitIO() { std::cin.tie(nullptr); std::ios_base::sync_with_stdio(false); std::cout << std::fixed << std::setprecision(30); } } init_io; }
#define mv_rec make_v(init, tail...)
template <typename T> T make_v(T init) { return init; }
template <typename T, typename... Tail> auto make_v(T init, size_t s, Tail... tail) { return V<decltype(mv_rec)>(s, mv_rec); }
#undef mv_rec
using namespace std;

#ifdef DEBUGGING
#else
#define DEBUG(...) 0
#define DEBUG_SEPARATOR_LINE 0
#endif

namespace graph {

using Node = ll;
using Weight = ll;
using Edge = pair<Node, Weight>;

template <bool Directed>
struct Graph : public VV<Edge> {
    using VV<Edge>::VV;

    void add_edge(Node f, Node t, Weight w = 1) {
        (*this)[f].emplace_back(t, w);
        if (!Directed) (*this)[t].emplace_back(f, w);
    }

    Graph<Directed> build_inv() const {
        Graph<Directed> ret(this->size());
        for (Node i = 0; i < this->size(); i++) {
            for (const Edge &e : (*this)[i]) {
                Node j;
                Weight w;
                tie(j, w) = e;
                if (!Directed && j < i) continue;
                ret.add_edge(j, i, w);
            }
        }

        return ret;
    }
};

}

void dfs(ll cur, ll pre, const graph::Graph<false> &graph, V<ll> &depth, ll d = 0) {
    depth[cur] = d;
    for (auto &&e : graph[cur]) {
        ll nxt;
        tie(nxt, ignore) = e;
        if (nxt == pre) continue;
        dfs(nxt, cur, graph, depth, d + 1);
    }
}

ll solve() {
    ll N, K;
    cin >> N >> K;
    if (N < K) return -1;
    graph::Graph<false> graph(N);
    for (ll i = 1; i < N; i++) {
        ll a, b;
        cin >> a >> b;
        a--; b--;
        graph.add_edge(a, b);
    }
    V<ll> depth(N);
    dfs(0, -1, graph, depth);
    sort(ALL(depth));
    ll ret = 0;
    for (ll i = 0; i < K; i++) ret += depth[i];
    return ret;
}

int main() {
    cout << solve() << endl;
    return 0;
}
0