【洛谷】P1064 金明的预算方案

题目地址：

https://www.luogu.com.cn/problem/P1064

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

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

如果要买归类为附件的物品，必须先买该附件所属的主件。每个主件可以有$0$个、$1$个或$2$个附件。每个附件对应一个主件，附件不再有从属于自己的附件。金明想买的东西很多，肯定会超过妈妈限定的$n$元。于是，他把每件物品规定了一个重要度，分为$5$等：用整数$1\sim 5$表示，第$5$等最重要。他还从因特网上查到了每件物品的价格（都是$10$元的整数倍）。他希望在不超过$n$元的前提下，使每件物品的价格与重要度的乘积的总和最大。设第$j$件物品的价格为$v_j$，重要度为$w_j$，共选中了$k$件物品，编号依次为$j_1,j_2,\dots ,j_k$，则所求的总和为：$v_{j_1}\times w_{j_1}+v_{j_2}\times w_{j_2}+\cdots +v_{j_k}\times w_{j_k}$。请你帮助金明设计一个满足要求的购物单。

输入格式：
第一行有两个整数，分别表示总钱数$n$和希望购买的物品个数$m$。
第$2$到第$(m+1)$行，每行三个整数，第$(i+1)$行的整数$v_i$，$p_i$，$q_i$ 分别表示第 $i$ 件物品的价格、重要度以及它对应的的主件。如果$q_i=0$，表示该物品本身是主件。

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

数据范围：
对于全部的测试点，保证$1\le n\le 3.2\times 10^4$，$1\le m\le 60$，$0\le v_i\le 10^4$，$1\le p_i\le 5$，$0\le q_i\le m$，答案不超过 $2\times 10^5$。

下面介绍一个比https://blog.csdn.net/qq_46105170/article/details/114298090更优的做法。

设$g[i]$是第$i$件物品的信息，即$g[i].v$是其价格，$g[i].w$是其总重要度（即其价格和重要度的乘积。下面简称”重要度“）。$f[k][s]$是只考虑前$k$个物品组合（第$i$个物品组合的意思是包含第i个物品及其附件。这里附件不被当成一个独立的物品，而只当成其主件的从属物品）的情况下，总价格不超过$s$的时候能得到的最大重要度。在计算$f[k][s]$的时候，我们按顺序考虑两种情况：
1、含第$k$个物品的主件，那么如果我们不取其附件的话，最大重要度就是$f[k-1][s-g[k].v]+g[k].w$。假设一开始$f[k]$这一行的值全是$0$，那么在以上式更新$f[k]$的时候，那些不为$0$的值意味着取了主件。接下来考虑第$k$个物品的每个附件，比如我们考虑第$1$个附件，我们从大到小枚举$s$，$s$从$m$枚举到$g[k].v+c[k][0].v$（因为体积要能同时放得下主件和附件），$c[k][0]$表示第$k$个物品的第$1$个附件，如果我们取这个附件，剩下物品产生的最大重要度就是$f[k][s-c[k][0].v]$，由于要取附件必须得取其主件，如何知道这个重要度是否取了主件呢？只需要看$f[k][s-c[k][0].v]$是否为$0$就行了，如果不是$0$，说明之前只枚举主件的时候更新过，从而意味着这是取了主件的情形。也就是说，如果发现$f[k][s-c[k][0].v]>0$，那么取附件就有可能更新答案，从而用$f[k][s-c[k][0].v]+c[k][0].w$来更新$f[k][s]$。由此可以看出，之所以$s$要从大到小更新，是为了计算的时候用的是更新前的值。
2、不含第$k$个物品的主件，那只需要直接用$f[k-1][s]$来更新$f[k][s]$即可。

注意，上面的两种情况一定要按顺序考虑，原因是我们利用了$f[k][s-c[k][0].v]>0$这个条件来判断主件有没有取，从而确定附件是否能取的。

代码如下：

#include <iostream>
#include <vector>
#define v first
#define w second
using namespace std;
using PII = pair<int, int>;

const int M = 32010, N = 70;
int n, m;
PII g[N];
vector<PII> son[N];
int f[N][M];

int main() {
  scanf("%d%d", &m, &n);
  for (int i = 1; i <= n; i++) {
    int a, b, c;
    scanf("%d%d%d", &a, &b, &c);
    if (!c) g[i] = {a, a * b};
    else son[c].push_back({a, a * b});
  }

  for (int i = 1; i <= n; i++) {
  	// 先考虑只取主件不取附件的情形
    for (int j = g[i].v; j <= m; j++)
      f[i][j] = f[i - 1][j - g[i].v] + g[i].w;
	
	// 枚举每个附件
    for (int k = 0; k < son[i].size(); k++)
      for (int j = m; j >= g[i].v + son[i][k].v; j--)
      	// 如果f[i][j - son[i][k].v] > 0，意味着主件已取，从而可以取附件
      	// 由于在这里这个条件等价于j - son[i][k].v >= g[i].v，所以这一行可以省略
        f[i][j] = max(f[i][j], f[i][j - son[i][k].v] + son[i][k].w);
	
	// 考虑不取主件的情况
    for (int j = 0; j <= m; j++)
      f[i][j] = max(f[i][j], f[i - 1][j]);
  }

  printf("%d\n", f[n][m]);
}

时空复杂度$O(nm)$。