HDU 4003 树形dp+分组背包

求K个机器人从同一点出发，遍历所有点所需的最小花费

注意到，对于某个节点为根节点的子树，如果需要遍历完再回来，一个机器人遍历的花费<=多个机器人的花费，所以一开始先用一个机器人遍历再回来，保证有一个解，如果有多余的机器人则选择最优的一条路而不需要返回，此时更新解，这就成经典的树形dp+分组背包了

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e4+6;
int dp[maxn][12],head[maxn],n,kk,s,num;
struct list {int u,v,w,next;}a[2*maxn];
void add(int u,int v,int w) {a[num]={u,v,w,head[u]};head[u]=num++;}
void dfs(int u,int fa)
{
    int i,j,k;
    for(i=head[u];~i;i=a[i].next)
    {
        int v=a[i].v,w=a[i].w;
        if(v==fa) continue;
        dfs(v,u);
        for(j=kk;j>=0;j--)
        {
            dp[u][j]+=dp[v][0]+2*w;  //dp[v][0]表示用一个机器人去走完所有子树，最后又回到v这个节点</span>
            for(k=1;k<=j;k++)
                dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[v][k]+k*w);
        }
    }
}
int main()
{
    int i,x,y,w,s;
    while(~scanf("%d%d%d",&n,&s,&kk))
    {
        memset(head,-1,sizeof(head));
        memset(dp,0,sizeof(dp));num=1;
        for(i=1;i<n;i++) scanf("%d%d%d",&x,&y,&w),add(x,y,w),add(y,x,w);
        dfs(s,0);
        printf("%d\n",dp[s][kk]);
    }
    return 0;
}