【luogu P1064】【ybtoj】【背包问题课堂过关】【DP】【例题5】金明的预算方案 & 金明的预算方案

ybtoj【背包问题课堂过关】【例题5】金明的预算方案
luogu 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

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

输出 #1

2200

说明/提示

数据规模与约定

对于全部的测试点，保证$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$。

---

解题思路

其实就是个01背包 搞得好像考试时我写出来了一样
普通的01背包有两个选择

• 1、不选，考虑下一个
• 2、选，更新最大值（f[j]=f[j-w[i]]+v[i]）
  现在变成5个选择
• 1、不选，考虑下一个
• 2、选主件，不选附件，更新最大值（f[j]=f[j-w[i]]+v[i]）
• 3、选主件，选附件1，更新最大值（f[j]=f[j-w[i]-nextw[i][1]]+v[i]+nextv[i][1]）
• 4、选主件，选附件2，更新最大值 f[j]=max(f[j],f[j-w[i]-nextw[i][2]]+v[i]+nextv[i][2]);
• 5、选主件，两个附件都买 （够rich），更新最大值 f[j]=max(f[j],f[j-w[i]-nextw[i][1]-nextw[i][2]]+v[i]+nextv[i][1]+nextv[i][2])
  为了让附件跟着主件，我们要做一个冗长~~~的预处理。好吧，也不是很长。。。

---

Code

#include<iostream>
#include<cstdio>
using namespace std;
int w[100],v[100],nextw[100][3],nextv[100][3],n,m,f[33000];
int main(){
	scanf("%d%d",&m,&n);
	for(int i=1;i<=n;i++){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		if(!z){//如果i是个主件
			w[i]=x;//就把TA存到主件的数组里
			v[i]=x*y;
		}else{//如果i是个附件，就存到附件的数组里
		    nextw[z][0]++;//next[z][0]用来统计z有几个附件
			nextw[z][nextw[z][0]]=x;
			nextv[z][nextw[z][0]]=x*y;	
		}
	}
	for(int i=1;i<=n;i++){
		if(w[i]==0)continue;//因为在i是附件的情况下w和v都跳过了TA，所以不统计附件
		for(int j=m;j>=w[i];j--){//上述五个选择
			f[j]=max(f[j],f[j-w[i]]+v[i]);
			if(j>=w[i]+nextw[i][1])
			   f[j]=max(f[j],f[j-w[i]-nextw[i][1]]+v[i]+nextv[i][1]);
			if(j>=w[i]+nextw[i][2])
			   f[j]=max(f[j],f[j-w[i]-nextw[i][2]]+v[i]+nextv[i][2]);
			if(j>=w[i]+nextw[i][1]+nextw[i][2])
			   f[j]=max(f[j],f[j-w[i]-nextw[i][1]-nextw[i][2]]+v[i]+nextv[i][1]+nextv[i][2]);
		}
	} 
	printf("%d",f[m]);
}