第三周练习题1

金属采集

问题

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

Input

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

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

Output

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

示例

输入

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

输出

3004

代码

#include <cstdio>
#include <cmath>
#include <vector>
#include <cstring>  
#define N 100005
  
using namespace std;
  
int n,s,K;
int dp[N][12];        //dp[p][m]:表示在以p为根的子树中停留m个机器人的花费。把一棵子树看作是一个整体。
typedef struct
 {
     int to;
     int w;
 }PP;
 
 vector<PP>bian[N];
 int vis[N];
 
 void dfs(int p)
 {
     vis[p]=1;
     int i;
     int next;
     int cost;
     int k;
     int remain;
     for(i=0;i<bian[p].size();i++)
     {
         next=bian[p][i].to;
         if(vis[next]==1) continue;
         cost=bian[p][i].w;
         dfs(next);
         for(k=K;k>=0;k--){        //这里要注意不能掉了0的情况
             dp[p][k]+=dp[next][0]+cost*2;        //子树son中没有停留机器人，那么意味全反回，最少是去一个所以最少反回一个
             for(remain=1;remain<=k;remain++){        //在子树son中停raim个机器人的情况
                 dp[p][k]=min(dp[p][k],dp[p][k-remain]+dp[next][remain]+remain*cost);
             }
         }
     }
 }
 
 int main()
 {
     int x,y,w;
     PP te;
     
     memset(dp,0,sizeof(dp) );
     memset(vis,0,sizeof(vis));
     int i;
     scanf("%d%d%d",&n,&s,&K);
     for(i=0;i<=n;i++){
         bian[i].clear();
     }    
     for(i=0;i<n-1;i++){
         scanf("%d%d%d",&x,&y,&w);
         te.w=w;
         te.to=y;
         bian[x].push_back(te);
         te.to=x;
         bian[y].push_back(te);
     }

     dfs(s);
     printf("%d",dp[s][K]);
     return 0;
 }