【ACWing】487. 金明的预算方案

题目地址:

https://www.acwing.com/problem/content/489/

金明今天很开心,家里购置的新房就要领钥匙了,新房里有一间金明自己专用的很宽敞的房间。更让他高兴的是,妈妈昨天对他说:“你的房间需要购买哪些物品,怎么布置,你说了算,只要不超过 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元(可以等于 N N N元)的前提下,使每件物品的价格与重要度的乘积的总和最大。设第 j j j件物品的价格为 v [ j ] v[j] v[j],重要度为 w [ j ] w[j] w[j],共选中了 k k k件物品,编号依次为 j 1 , j 2 , … , j k j_1,j_2,…,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]∗w[j_1]+v[j_2]∗w[j_2]+…+v[j_k]∗w[j_k] v[j1]w[j1]+v[j2]w[j2]++v[jk]w[jk](其中 ∗ * 为乘号)请你帮助金明设计一个满足要求的购物单。

输入格式:
输入文件的第 1 1 1行,为两个正整数,用一个空格隔开: N   m N\ m N m,其中 N N N表示总钱数, m m m为希望购买物品的个数。从第 2 2 2行到第 m + 1 m+1 m+1行,第 j j j行给出了编号为 j − 1 j-1 j1的物品的基本数据,每行有 3 3 3个非负整数 v , p , q v, p, q v,p,q,其中 v v v表示该物品的价格, p p p表示该物品的重要度( 1 ∼ 5 1\sim 5 15), q q q表示该物品是主件还是附件。如果 q = 0 q=0 q=0,表示该物品为主件,如果 q > 0 q>0 q>0,表示该物品为附件, q q q是所属主件的编号(指的是第一次遇到这个主件的时候的计数,从 1 1 1开始数)。

输出格式:
输出文件只有一个正整数,为不超过总钱数的物品的价格与重要度乘积的总和的最大值( < 200000 <200000 <200000)。

数据范围:
N < 32000 , m < 60 , v < 10000 N<32000,m<60,v<10000 N<32000,m<60,v<10000

更好的做法参考https://blog.csdn.net/qq_46105170/article/details/125026481

可以看成是分组背包问题,每个物品体积就是 v i v_i vi,价值就是 v i × p i v_i\times p_i vi×pi。每个主件及其所有附件打包成一起看成一组,每组的选择是,可以买其主件与附件的任意搭配(比如其有 u u u个附件,那么搭配的方案数就是 2 u 2^u 2u种)。用动态规划来做即可,设 f [ i ] [ j ] f[i][j] f[i][j]是只考虑前 i i i组物品的情况下,总成本不超过 j j j时的最大价值,那么可以按照第 i i i组选哪个来分类(这里的“选哪个”包含第 i i i组物品的所有搭配,比如不选第 i i i组,只选主件,只选主件和第一附件,等等),如果不选,那么最大价值就是 f [ i − 1 ] [ j ] f[i-1][j] f[i1][j],如果选,那么最大价值就是 f [ i − 1 ] [ j − v i ] + v i × p i f[i-1][j-v_i]+v_i\times p_i f[i1][jvi]+vi×pi(这里要遍历所有附件的搭配情况),两者取大即可。代码如下:

#include <iostream>
#include <vector>
using namespace std;

typedef pair<int, int> PII;

const int N = 70, M = 32010;

int n, m;
// 存主件的价格和收益
PII master[N];
// servant[i]存i号主件的价格和收益
vector<PII> servant[N];
int f[M];

int main() {
    cin >> m >> n;
    for (int i = 1; i <= n; i++) {
        int v, w, q;
        cin >> v >> w >> q;
        // q是0的话,当前物品是主件
        if (!q) master[i] = {v, v * w};
        // 否则是附件,价格是v,收益是v * w
        else servant[q].push_back({v, v * w});
    }

    for (int i = 1; i <= n; i++) 
    	// 这是0-1背包问题,体积要从大到小遍历
        for (int j = m; j >= 0; j--) {
            auto &sv = servant[i];
            // 枚举所有附件的搭配情况,用状态压缩的办法枚举
            for (int k = 0; k < 1 << sv.size(); k++) {
                int v = master[i].first, w = master[i].second;
                for (int u = 0; u < sv.size(); u++)
                    if (k >> u & 1) {
                        v += sv[u].first;
                        w += sv[u].second;
                    }
                    
                if (j >= v) f[j] = max(f[j], f[j - v] + w);
            }
        }

    cout << f[m] << endl;

    return 0;
}

时间复杂度 O ( N m ) O(Nm) O(Nm),空间 O ( m ) O(m) O(m)

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值