hdu4003Find Metal Mineral(树形DP）

4003

思维啊 dp[i][j]表示当前I节点停留了j个机器人 那么它与父亲的关系就有了 那条边就走了j遍

dp[i][j] = min(dp[i][j],dp[child][g]+dp[i][j-g]+g*w[i][child] );

 #include <iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<stdlib.h>
 #include<vector>
 #include<cmath>
 #include<queue>
 using namespace std;
 #define N 10010
 #define LL __int64
 struct node
 {
     int u,v,w,next;
 }ed[N<<];
 int head[N],t,k;
 int dp[N][];
 void init()
 {
     t = ;
     memset(head,-,sizeof(head));
 }
 void add(int u,int v,int w)
 {
     ed[t].u = u;
     ed[t].v = v;
     ed[t].w = w;
     ed[t].next = head[u];
     head[u] = t++;
 }
 void dfs(int pre,int u)
 {
     int i,j;
     for(i = head[u] ; i!=- ; i = ed[i].next)
     {
         int v = ed[i].v;
         if(v==pre) continue;
         dfs(u,v);
         for(j = k ; j >=  ; j --)
         {
             dp[u][j]+=dp[v][]+*ed[i].w;
             for(int g =  ; g <= j ; g++)
             {
                 dp[u][j] = min(dp[u][j],dp[u][j-g]+dp[v][g]+g*ed[i].w);
             }
         }
     }
 }
 int main()
 {
     int i,n,s;
     while(scanf("%d%d%d",&n,&s,&k)!=EOF)
     {
         memset(dp,,sizeof(dp));
         init();
         for(i =  ; i < n ; i++)
         {
             int u,v,w;
             scanf("%d%d%d",&u,&v,&w);
             add(u,v,w);
             add(v,u,w);
         }
         dfs(-,s);
         printf("%d\n",dp[s][k]);
     }
     return ;
 }