树形背包 金属采集

树形背包 金属采集

时间限制：1.0s 内存限制：256.0MB

问题描述
人类在火星上发现了一种新的金属！这些金属分布在一些奇怪的地方，不妨叫它节点好了。一些节点之间有道路相连，所有的节点和道路形成了一棵树。一共有 n 个节点，这些节点被编号为 1~n 。人类将 k 个机器人送上了火星，目的是采集这些金属。这些机器人都被送到了一个指定的着落点， S 号节点。每个机器人在着落之后，必须沿着道路行走。当机器人到达一个节点时，它会采集这个节点蕴藏的所有金属矿。当机器人完成自己的任务之后，可以从任意一个节点返回地球。当然，回到地球的机器人就无法再到火星去了。我们已经提前测量出了每条道路的信息，包括它的两个端点 x 和 y，以及通过这条道路需要花费的能量 w 。我们想花费尽量少的能量采集所有节点的金属，这个任务就交给你了。

输入格式

第一行包含三个整数 n, S 和 k ，分别代表节点个数、着落点编号，和机器人个数。
接下来一共 n-1 行，每行描述一条道路。一行含有三个整数 x, y 和 w ，代表在 x 号节点和 y 号节点之间有一条道路，通过需要花费 w 个单位的能量。所有道路都可以双向通行。

输出格式

输出一个整数，代表采集所有节点的金属所需要的最少能量。

样例输入

6 1 3
1 2 1
2 3 1
2 4 1000
2 5 1000
1 6 1000

样例输出

3004

样例说明
所有机器人在 1 号节点着陆。
第一个机器人的行走路径为 1->6 ，在 6 号节点返回地球，花费能量为1000。
第二个机器人的行走路径为 1->2->3->2->4 ，在 4 号节点返回地球，花费能量为1003。
第一个机器人的行走路径为 1->2->5 ，在 5 号节点返回地球，花费能量为1001。
数据规模与约定
本题有10个测试点。
对于测试点 1~2 ， n <= 10 ， k <= 5 。
对于测试点 3 ， n <= 100000 ， k = 1 。
对于测试点 4 ， n <= 1000 ， k = 2 。
对于测试点 5~6 ， n <= 1000 ， k <= 10 。
对于测试点 7~10 ， n <= 100000 ， k <= 10 。
道路的能量 w 均为不超过 1000 的正整数。

#include <iostream>
using namespace std;
const int maxn = 6 * 1e3 + 10;
int n,s,k,num;
bool vis[maxn];
int dp[maxn][11],head[maxn]; //dp[m][k]表示以m节点为根的树，来k个机器人花费的最少能量

struct node {
    int to, next,weight;
} ed[maxn];

void add(int x, int y,int z) {
    ed[++num].next = head[x];
    ed[num].to = y;
    ed[num].weight = z;
    head[x] = num;
}

void tree_dp(int x) {
    vis[x] = true;
    for (register int i = head[x]; i; i = ed[i].next) {
        int v = ed[i].to, weight = ed[i].weight;
        if (vis[v]) continue;
        tree_dp(v);
        //由于 dp[x][j] 由dp[x][j - m] + dp[v][m]更新得到，那么就需要保证更新 dp[x][j] 的时候 dp[v][m] 没有被更新，由于 m 是一个正整数，所以我们倒序枚举 k 即可
        for (register int j = k; j >= 0; --j) {
            //子树中没有停留机器人，那么意味全反回，最少是去一个所以最少反回一个,等于子节点存0个机器人+一个机器人从父节点到子节点，再从子节点到父节点（即2*w）。
            dp[x][j] += dp[v][0] + weight * 2;
            //子树中停留（1~j)个机器人
            for (register int m = 1; m <= j; ++m) {
                dp[x][j] = min(dp[x][j], dp[x][j - m] + dp[v][m] + m * weight);
            }
        }
    }
}
int main() {
    cin >> n >> s >> k;
    for(register int i=1,a,b,c;i<n;++i) {
        cin>>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
    }
    tree_dp(s);
    cout << dp[s][k] << endl;
    return 0;
}