J - Find Metal Mineral HDU - 树形DP+背包

ID_BePosit

于 2018-10-28 21:12:15 发布

阅读量161

点赞数

分类专栏： DP

本文链接：https://blog.csdn.net/BePosit/article/details/83478298

版权

DP 专栏收录该内容

105 篇文章 0 订阅

订阅专栏

J - Find Metal Mineral
HDU - 4003
题意：k个机器人到S点，问机器人找完所有金矿后，最短的移动距离；如果只有一个机器人，
他就要找遍所有金矿就会走重复的路，直到找到最后一个矿；如果有两个机器人，他俩就可以分头行动；
思路：dp[i][j]=min(dp[i][j-k]+dp[son][k] | 0<=k<=j)设dp[i][j]表示i节点分配j个机器人的最短移动距离
0并不表示不分配机器人了，而是只有一个机器人分配过来，并且最终返回父节点；

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define maxn 10086
int n,k,s,x,y,z;
int head[maxn],tot;
int dp[maxn][15];
struct node
{
    int v,w,to;
} edge[maxn*2];
void add(int x,int y,int z)
{
    edge[++tot].v=y;
    edge[tot].w=z;
    edge[tot].to=head[x];
    head[x]=tot;
    edge[++tot].v=x;
    edge[tot].w=z;
    edge[tot].to=head[y];
    head[y]=tot;
}
void dfs(int root,int pre)
{
    for(int i=head[root]; i!=-1; i=edge[i].to)
    {
        int son=edge[i].v;
        int dis=edge[i].w;
        if(son==pre)continue;
        dfs(son,root);
        for(int j=k; j>=0; j--)
        {
            dp[root][j]+=dp[son][0]+dis*2;
            for(int p=1; p<=j; p++)
                dp[root][j]=min(dp[root][j],dp[root][j-p]+dp[son][p]+dis*p);
        }
    }
}
int main()
{
    while(~scanf("%d%d%d",&n,&s,&k))
    {
        tot=0;
        memset(head,-1,sizeof(head));
        memset(dp,0,sizeof(dp));
        for(int i=1; i<n; i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,z);
        }
        dfs(s,s);
        printf("%d\n",dp[s][k]);
    }
    return 0;
}