有依赖的背包问题(树形dp)

有依赖的背包问题

有 N 个物品和一个容量是 V 的背包。

物品之间具有依赖关系,且依赖关系组成一棵树的形状。如果选择一个物品,则必须选择它的父节点。

如下图所示:
QQ图片20181018170337.png

如果选择物品5,则必须选择物品1和2。这是因为2是5的父节点,1是2的父节点。

每件物品的编号是 i,体积是 vi,价值是 wi,依赖的父节点编号是 pi。物品的下标范围是 1…N。

求解将哪些物品装入背包,可使物品总体积不超过背包容量,且总价值最大。

输出最大价值。

输入格式

第一行有两个整数 N,V,用空格隔开,分别表示物品个数和背包容量。

接下来有 N 行数据,每行数据表示一个物品。
第 i 行有三个整数 vi,wi,pi,用空格隔开,分别表示物品的体积、价值和依赖的物品编号。
如果 pi=−1,表示根节点。 数据保证所有物品构成一棵树。

输出格式

输出一个整数,表示最大价值。

数据范围

1≤N,V≤100
1≤vi,wi≤100

父节点编号范围:

  • 内部结点:1≤pi≤N;
  • 根节点 pi=−1;

输入样例

5 7
2 3 -1
2 2 1
3 5 1
4 7 2
3 6 2

输出样例:

11

依赖背包问题核心在于将暴力枚举子树的选举情况转换成对体积的枚举

首先枚举每一个子树,然后把子树计算出来,然后将当前树的各个子树作为每一个物品组,这些物品是按照体积来决策的,所以再枚举体积(因为x必须选,所以要将这一部分的体积留出来),最后枚举决策,也就是枚举当前背包容量。然后更新的方法:先回忆一下之前的普通分组背包中是以是否选择当前物品来决策,因此体积就是减去当前物品的体积进行更新f[j-v[k]]+w[k]。但是这个是按照体积决策,也就是说不是对于一个物品组中的物品选还是不选,而是为这个物品组分配多少空间的问题,因此如果分配k的话,那就是f[x][j]=max(f[x][j],f[x][j−k]+f[y][k])

#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
const int N = 105;
vector<int>son[N];
int v[N],w[N],f[N][N];
int n,m;
void dp(int x){
    for(int i=0;i<son[x].size();i++){
        int y=son[x][i];
        dp(y);
        for(int j=m-v[x];j>=0;j--){//背包体积
            for(int k=0;k<=j;k++)
                f[x][j]=max(f[x][j],f[x][j-k]+f[y][k]);
        }
    }
    for(int i=m;i>=v[x];i--)
        f[x][i]=f[x][i-v[x]]+w[x];
    for(int i=0;i<v[x];i++)
        f[x][i]=0;
}
int main(){
    int root;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        int p;
        scanf("%d%d%d",&v[i],&w[i],&p);
        if(p==-1) root=i;
        else son[p].push_back(i);
    }
    dp(root);
    printf("%d",f[root][m]);
    return 0;
}

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值