【题解】hdu4003 树形DP

题目链接
dp[i][j]表示以i为根的子树放j个机器人所产生的最小权值和,dp[i][0]表示放了一个机器人遍历i的子树然后返回i所产生的权值和。采用邻接表

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e5+10;
#define INF 0x3f3f3f3f
struct node{
    int next,v,w;
}edge[N];
int n,s,k;//金属矿物，着陆点，机器人数量 
int tot,head[N];
int dp[N][15];//dp[i][j]表示以 i为根的子树，放j个机器人所需要的权值和 
void addadge(int u,int v,int w)
{
    edge[tot].v=v;
    edge[tot].w=w;
    edge[tot].next=head[u];
    head[u]=tot++;
}
void dfs(int u,int pre)
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        if(v==pre)continue;
        dfs(v,u);
        for(int j=k;j>=0;j--)
        {
            dp[u][j]+=dp[v][0]+2*edge[i].w;
            //dp[i][0]表示放1个机器人，遍历节点后回到i节点
            for(int l=1;l<=j;l++)
            dp[u][j]=min(dp[u][j],dp[u][j-l]+dp[v][l]+l*edge[i].w);
        }
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    int i,u,v,w;
    while(~scanf("%d%d%d",&n,&s,&k))
    {
        tot=0;
        memset(head,-1,sizeof(head));
        for(i=1;i<n;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            addadge(u,v,w);
            addadge(v,u,w);
        }
        memset(dp,0,sizeof(dp));
        dfs(s,0);
        printf("%d\n",dp[s][k]);
    }
    return 0;
}