hdu4003 Find Metal Mineral(树形dp+分组背包思想)

题目

n(n<=1e4)个点的树，每条边有一个边权w(w<=1e4)，通过其的代价为其权值

你有k(k<=10)个机器人，所有机器人从给定的源点s(1<=s<=n)出发，你可以控制机器人的移动

求机器人访问所有点时，其最小的边权代价总和，输出代价总和

1<=N<=10000, 1<=S<=N, 1<=k<=10, 1<=x, y<=N, 1<=w<=10000.

思路来源

https://www.cnblogs.com/kuangbin/archive/2012/08/29/2661928.html

题解

显然树形dp，只考虑父亲u和儿子v，u能去访问v当且仅当u处还有机器人，

因此，考虑状态dp[u][x][y]为根节点为u时，访问了其子树x个节点，留在根节点y个机器人的最小代价，

然而发现状态转移会爆，后来参考题解发现第二维多余，因为每棵子树都要求全部访问，

于是dp[i][j]表示在i的子树里(不含i)有j个机器人未返回根节点i，且遍历完所有的点的最小代价

dp[i][0]表示留下了0个，即所有机器人都回到了根i

然后考虑转移，u->v派机器人的时候，不同的决策是在v子树里留下0个，1个，…，j个机器人

以上决策互斥，且必须选一个，这就用到了分组背包的思想

①v子树里留0个机器人，发现如果用一些机器人从u访问v子树所有节点并均回到u，

超过一个机器人的方案是不如一个机器人优的，考虑每条带叶子节点的链都需要被访问，

如果派出>=2个机器人，叶子节点的覆盖次数不变，分支节点的覆盖次数增加，显然不可取

则只用一个机器人，考虑u->v的贡献为往返的2*w

如果所有的机器人在之前的子树都留下，怎么办，

可以先用一个机器人走v这棵子树并不留，然后再留在之前的子树里

②v子树里留l个机器人，则u之前的子树里留j-l个机器人（l个机器人已经回到u），

只用考虑这l个机器人，往下走到v的这l*w的贡献

 

分组背包的原版本是可以不选的，必须选一个的时候，

我们就先必选第一个，然后尝试用后续方案替换

 

虽然核心代码就三四行，然而理解了快一个小时，我可真sb啊orz

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
typedef pair<int,int> P;
#define fi first
#define se second
const int N=1e4+10,K=11;
//只考虑u->v这一步的贡献
int n,s,k,u,v,w;
int dp[N][K];
//dp[i][j]表示在i的子树里(不含i)留下了j个机器人 且遍历完所有的点的最小代价
//dp[i][0]表示留下了0个，即都回到了根i
//显然 如果v的子树里不留机器人 仅用一个机器人去遍历v的子树最优
vector<P>e[N];
void dfs(int u,int fa){
    for(int i=0;i<e[u].size();++i){
        int v=e[u][i].fi,w=e[u][i].se;
        if(v==fa)continue;
        dfs(v,u);
        for(int j=k;j>=0;--j){
            dp[u][j]+=dp[v][0]+2*w;//u->v v->u 只用一个机器人访问v子树 并且回到v节点
            for(int l=1;l<=j;++l){//在v的子树里留下[1,j]个机器人 u->v l个机器人下放
                dp[u][j]=min(dp[u][j],dp[u][j-l]+dp[v][l]+l*w);
            }
        }
    }
}
int main(){
    while(~scanf("%d%d%d",&n,&s,&k)){
        memset(dp,0,sizeof dp);
        for(int i=1;i<=n;++i){
            e[i].clear();
        }
        for(int i=1;i<n;++i){
            scanf("%d%d%d",&u,&v,&w);
            e[u].push_back(P(v,w));
            e[v].push_back(P(u,w));
        }
        dfs(s,-1);
        printf("%d\n",dp[s][k]);
    }
    return 0;
}