有依赖的背包,狭义(二进制枚举),广义(树形dp)

[NOIP2006 提高组] 金明的预算方案

题目描述

金明今天很开心,家里购置的新房就要领钥匙了,新房里有一间金明自己专用的很宽敞的房间。更让他高兴的是,妈妈昨天对他说:“你的房间需要购买哪些物品,怎么布置,你说了算,只要不超过 n n n 元钱就行”。今天一早,金明就开始做预算了,他把想买的物品分为两类:主件与附件,附件是从属于某个主件的,下表就是一些主件与附件的例子:

主件附件
电脑打印机,扫描仪
书柜图书
书桌台灯,文具
工作椅

如果要买归类为附件的物品,必须先买该附件所属的主件。每个主件可以有 0 0 0 个、 1 1 1 个或 2 2 2 个附件。每个附件对应一个主件,附件不再有从属于自己的附件。金明想买的东西很多,肯定会超过妈妈限定的 n n n 元。于是,他把每件物品规定了一个重要度,分为 5 5 5 等:用整数 1 ∼ 5 1 \sim 5 15 表示,第 5 5 5 等最重要。他还从因特网上查到了每件物品的价格(都是 10 10 10 元的整数倍)。他希望在不超过 n n n 元的前提下,使每件物品的价格与重要度的乘积的总和最大。

设第 j j j 件物品的价格为 v j v_j vj,重要度为 w j w_j wj,共选中了 k k k 件物品,编号依次为 j 1 , j 2 , … , j k j_1,j_2,\dots,j_k j1,j2,,jk,则所求的总和为:

v j 1 × w j 1 + v j 2 × w j 2 + ⋯ + v j k × w j k v_{j_1} \times w_{j_1}+v_{j_2} \times w_{j_2}+ \dots +v_{j_k} \times w_{j_k} vj1×wj1+vj2×wj2++vjk×wjk

请你帮助金明设计一个满足要求的购物单。

输入格式

第一行有两个整数,分别表示总钱数 n n n 和希望购买的物品个数 m m m

2 2 2 到第 ( m + 1 ) (m + 1) (m+1) 行,每行三个整数,第 ( i + 1 ) (i + 1) (i+1) 行的整数 v i v_i vi p i p_i pi q i q_i qi 分别表示第 i i i 件物品的价格、重要度以及它对应的的主件。如果 q i = 0 q_i=0 qi=0,表示该物品本身是主件。

输出格式

输出一行一个整数表示答案。

样例 #1

样例输入 #1

1000 5
800 2 0
400 5 1
300 5 1
400 3 0
500 2 0

样例输出 #1

2200

提示

数据规模与约定

对于全部的测试点,保证 1 ≤ n ≤ 3.2 × 1 0 4 1 \leq n \leq 3.2 \times 10^4 1n3.2×104 1 ≤ m ≤ 60 1 \leq m \leq 60 1m60 0 ≤ v i ≤ 1 0 4 0 \leq v_i \leq 10^4 0vi104 1 ≤ p i ≤ 5 1 \leq p_i \leq 5 1pi5 0 ≤ q i ≤ m 0 \leq q_i \leq m 0qim,答案不超过 2 × 1 0 5 2 \times 10^5 2×105


有依赖的背包简单版,或者直接叫做分组背包。

你能发现下边的代码哪里错了吗

时间复杂度 O ( n m ) O(nm) O(nm)

int n,m;
struct good{
    int v,w;
};
vector<good>g[70];
int f[40010];
void solve(){
    cin>>m>>n;
    fo(i,1,n){
        int v,w,s;cin>>v>>w>>s;
        if(!s){
            g[i].pb({v,w});
        }
        else{
            g[s].pb({v,w});
        }
    }
    for(int j=m;j>=0;j--){
        for(int i=1;i<=n;i++){
        	if(!g[i].size())continue;
            auto p = *g[i].begin();
            // debug(p.v);
            for(int k=0;k<((g[i].size()-1)<<1);k++){
            	// debug(k);
                if(!k){
                    if(j-p.v>=0)
                        f[j] = max(f[j],f[j-p.v]+p.w*p.v);
                }
                else if(k==1 || k==2){
                    auto x = g[i][k];
                    if(j-x.v-p.v>=0)
                        f[j] = max(f[j],f[j-x.v-p.v]+x.w*x.v+p.w*p.v);
                }
                else{
                    auto x = g[i][1];
                    auto y = g[i][2];
                    if(j-x.v-y.v-p.v>=0)
                        f[j] = max(f[j],f[j-x.v-y.v-p.v]+x.w*x.v+y.w*y.v+p.w*p.v);
                }
            }
        }
    }
    cout<<f[m]<<endl;
}

int main(){
    solve();
    return 0;
}

解密:

if(!k){
    if(j-p.v>=0)
        f[j] = max(f[j],f[j-p.v]+p.w*p.v);
}
这句需要放到外边,不然循环不到qwq

有依赖的背包问题,树上背包

一个讲得非常不错的连接

抽象出模型,类比树形dp, f [ u ] [ j ] f[u][j] f[u][j] 表示以 u u u 节点为根节点,花费 j j j 所能得到的最大价值。

考虑每个儿子选还是不选,考虑到儿子的儿子的体积一定是 1 1 1 的倍数,所以对 u u u 节点的所有儿子做分组背包。

先 dfs 进去,回溯的时候做分组背包。

注意初始化

for(int i=v[u];i<=m;i++){
    f[u][i] = w[u];
}
int n,m;
int h[110],e[210],ne[210],idx;
int v[110],w[110];
int f[110][110];// f[u][j] 表示 u 为根的子树花费 j 取得的最大值
int root;
void add(int a,int b){
    e[idx]=b;
    ne[idx]=h[a];
    h[a]=idx++;
}

void dfs(int u,int fa){
    for(int i=v[u];i<=m;i++){
        f[u][i] = w[u];
    }
    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        if(j==fa)continue;
        dfs(j,u);
        for(int k=m;k>=v[u];k--){
            for(int l=0;l<=k-v[u];l++){
                f[u][k] = max(f[u][k],f[u][k-l]+f[j][l]);
            }
        }
    }
}

void solve(){
    memset(h,-1,sizeof(h));
    cin>>n>>m;
    fo(i,1,n){
        int x;
        cin>>v[i]>>w[i]>>x;
        if(x!=-1){
            add(x,i);
            add(i,x);
        }
        else root = i;
    }
    dfs(root,-1);
    cout<<f[root][m]<<endl;
}

int main(){
    solve();
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值