hdu6060[贪心+dfs] 2017多校3

/* hdu6060[贪心+dfs] 2017多校3*/
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
int n, k, u, v, c;
struct Edge {
    LL cost;
    int to;
    Edge(int C = , int T = ): cost(C), to(T) {}
};
vector<Edge> G[];
int son[];
LL ans = ;
void dfs(int u, int fa) {
    son[u]++;
    for (int i = ; i < (int)G[u].size(); i++) {
        Edge e = G[u][i];
        if (e.to == fa) continue;
        dfs(e.to, u);
        son[u] += son[e.to];
        ans += min(son[e.to], k) * e.cost;
    }
}
void init() {
    ans = ;
    for (int i = ; i <= n; i++) {
        G[i].clear();
    }
    memset(son, , sizeof(son));
}
void solve() {
    dfs(, -);
    printf("%lld\n", ans);
}
int main() {
    while (~scanf("%d%d", &n, &k)) {
        init();
        for (int i = ; i < n - ; i++) {
            scanf("%d%d%lld", &u, &v, &c);
            G[u].push_back(Edge(c, v));
            G[v].push_back(Edge(c, u));
        }
        solve();
    }
    return ;
}