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

题目地址：

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

金明今天很开心，家里购置的新房就要领钥匙了，新房里有一间金明自己专用的很宽敞的房间。更让他高兴的是，妈妈昨天对他说：“你的房间需要购买哪些物品，怎么布置，你说了算，只要不超过$N$元钱就行”。今天一早，金明就开始做预算了，他把想买的物品分为两类：主件与附件，附件是从属于某个主件的，如果要买归类为附件的物品，必须先买该附件所属的主件。每个主件可以有$0$个、$1$个或$2$个附件。附件不再有从属于自己的附件。金明想买的东西很多，肯定会超过妈妈限定的$N$元。于是，他把每件物品规定了一个重要度，分为$5$等：用整数$1\sim 5$表示，第$5$等最重要。他还从因特网上查到了每件物品的价格（都是$10$元的整数倍）。他希望在不超过$N$元（可以等于$N$元）的前提下，使每件物品的价格与重要度的乘积的总和最大。设第$j$件物品的价格为$v[j]$，重要度为$w[j]$，共选中了$k$件物品，编号依次为$j_1,j_2,\dots ,j_k$，则所求的总和为：$v[j_1]\ast w[j_1]+v[j_2]\ast w[j_2]+\dots +v[j_k]\ast w[j_k]$（其中$\ast$为乘号）请你帮助金明设计一个满足要求的购物单。

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

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

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

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

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

#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(Nm)$，空间$O(m)$。