[HAOI2015]树上染色（树上背包）

题目描述

有一棵点数为 N 的树，树边有边权。给你一个在 0~ N 之内的正整数 K ，你要在这棵树中选择 K个点，将其染成黑色，并将其他 的N-K个点染成白色 。 将所有点染色后，你会获得黑点两两之间的距离加上白点两两之间的距离的和的受益。问受益最大值是多少。

输入格式

第一行包含两个整数 N, K 。接下来 N-1 行每行三个正整数 fr, to, dis ， 表示该树中存在一条长度为 dis 的边 (fr, to) 。输入保证所有点之间是联通的。k<=n<=2000

输出格式

输出一个正整数，表示收益的最大值。

思路

标准的树上背包的题目。

当我们考虑两两点之间的距离和时，我们通常考虑边，一个边对答案的贡献，就是边两边黑白两种点的个数乘积，这样计算贡献就会方便很多。

设dp(i,j)为以i为根节点的子树上，选择j个点染色，对答案的贡献的最大值。

那么问题就变成了树形背包了，对于我们在访问的节点u，以及他的一个直接相连的子节点v，我们把边(u,v,cost)当作价值，给子节点v的子树分配黑节点数当作体积，这样是不是很像一个背包问题了：分给子树v的黑节点越多，占用的体积越大，但是我们得到边的价值，边的价值怎么计算呢？