HDU 6060 - RXD and dividing | 2017 Multi-University Training Contest 3

/*
HDU 6060 - RXD and dividing [ 分析,图论 ]  |  2017 Multi-University Training Contest 3
题意：
给一个 n 个节点的树，要求将 2-n 号节点分成 k 部分，然后将每一部分加上节点 1，
每一个子树的 val 为最小斯坦纳树，求总的最大 val
分析：
考虑每条边下面所在的子树，大小为num
由于该子树至多被分成 k 块，故该边最多贡献 k 次，贡献次数当然是越多越好
所以每条边的贡献为 w * min(k, num)
*/
#include <bits/stdc++.h>
using namespace std;
const int N = 1000005;
#define LL long long
struct Edge {
int to, w, next;
}edge[N<<1];
int head
, tot;
void init() {
memset(head, -1, sizeof(head));
tot = 0;
}
void addedge(int u, int v, int w)
{
edge[tot].to = v; edge[tot].w = w;
edge[tot].next = head[u];
head[u] = tot++;
}
int sum
, w
;
int dfs(int u, int pre)
{
sum[u] = 1;
for (int i = head[u]; i != -1; i = edge[i].next)
{
int v = edge[i].to;
if (v == pre) continue;
sum[u] += dfs(v, u);
w[v] = edge[i].w;
}
return sum[u];
}
int k, n;
int main()
{
while (~scanf("%d%d", &n, &k))
{
init();
int a, b, c;
for (int i = 1; i < n; i++)
{
scanf("%d%d%d", &a, &b, &c);
addedge(a, b, c);
addedge(b, a, c);
}
dfs(1, 1);
LL ans = 0;
for (int i = 2; i <= n; i++)
{
ans += (LL)min(sum[i], k) * w[i];
}
printf("%lld\n", ans);
}
}