金属采集

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

Input

第一行包含三个整数 n, S 和 k ，分别代表节点个数、着落点编号，和机器人个数。

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

#include <iostream>
#include <cstring>

using namespace std;

#define N 100001

struct edge
{
    int to;
    int weight;
    int next;
};

int tree[N];

edge edges[2*N];

int d[N][12];

int K;

int ep = 0;
void add_edge(int x, int y, int w)
{
    edges[ep].to = y;
    edges[ep].weight = w;
    edges[ep].next = tree[x];
    tree[x] = ep++;
}

void dfs(int r, int p)
{
    for(int cur = tree[r]; cur != -1; cur=edges[cur].next)
    {
        int child = edges[cur].to;
        if(child == p)
            continue;
        dfs(child, r);
        for(int i = K; i >= 0; i--)      // d[r][i] = ?
        {
            d[r][i] += d[child][0]+edges[cur].weight*2;

            for(int j = 1; j <= i; j++)
            {
                if(d[r][i] > d[child][j] + d[r][i-j] + j*edges[cur].weight)
                {
                    d[r][i] = d[child][j] + d[r][i-j] + j*edges[cur].weight;
                }
            }
        }

    }
}

int main()
{
    memset(tree, -1, sizeof(tree));
    int n, S;
    std::cin >> n >> S >> K;
    for(int i = 0; i < n-1; i++)
    {
        int a, b ,c;
        std::cin >> a >> b >> c;
        add_edge(a, b, c);
        add_edge(b, a, c);
    }

    dfs(S, -1);
    std::cout << d[S][K] << std::endl;
}