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

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

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

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

设第 jj 件物品的价格为 v_jvj​，重要度为w_jwj​，共选中了 kk 件物品，编号依次为 j_1,j_2,\dots,j_kj1​,j2​,…,jk​，则所求的总和为：

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​​。

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

输入格式

第一行有两个整数，分别表示总钱数 nn 和希望购买的物品个数 mm。

第 22 到第 (m + 1)(m+1) 行，每行三个整数，第 (i + 1)(i+1) 行的整数 v_ivi​，p_ipi​，q_iqi​ 分别表示第 ii 件物品的价格、重要度以及它对应的的主件。如果 q_i=0qi​=0，表示该物品本身是主件。

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

2200

说明/提示

数据规模与约定

对于全部的测试点，保证 1 \leq n \leq 3.2 \times 10^41≤n≤3.2×104，1 \leq m \leq 601≤m≤60，0 \leq v_i \leq 10^40≤vi​≤104，1 \leq p_i \leq 51≤pi​≤5，0 \leq q_i \leq m0≤qi​≤m，答案不超过 2 \times 10^52×105。

思路：一个主件有0-2个附件，买附件的话一定要买主件，买主件不一定需要买附件，那么我们就从买主件i开始入手，有四种情况：1.不够买主件i 2.只能买主件i 3.能买主件i和附件1 4.能买主件i和附件2。那么我们就根据这四种情况，用一个二维数组v[i][j]来记录编号为i的j主件或附件的价格，当j==0时是记录编号为i的主件加附件的个数，当j==1时表示的是主件，j==2表示的是附件1，j==3表示的是附件2。然后再用一个二维数组p[i][j]来表示编号为i的j主件或附件的价格×重要值，p[i][j]表示编号为i，当j==0时表示的编号为i的主件+附件的数量，当j==1时表示主件的价格×重要值，当j==2时表示附件1的价格×重要值，j==3时表示附件2的价格×重要值。

然后用01背包问题一个个枚举价格是否大于那四种情况然后更新就可以了。

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;
int n,m;
int f[32005];
int v[65][4];
int p[65][4];
int main(){
	cin>>m>>n;
	for(int i=1;i<=n;i++){
		int x,y,z;
		cin>>x>>y>>z;
		if(z==0){
			v[i][++v[i][0]]=x;
			p[i][++p[i][0]]=x*y;
		}else{
			v[z][++v[z][0]]=x;
			p[z][++p[z][0]]=x*y;
		}
	}
	for(int i=1;i<=n;i++){
		if(v[i][0]==0)continue;
		for(int j=m;j>=v[i][1];j--){
			f[j]=max(f[j],f[j-v[i][1]]+p[i][1]);
			if(v[i][0]>=2&&j>=v[i][1]+v[i][2]){
				f[j]=max(f[j],f[j-v[i][1]-v[i][2]]+p[i][1]+p[i][2]);
			}
			if(v[i][0]>=3&&j>=v[i][1]+v[i][3]){
				f[j]=max(f[j],f[j-v[i][1]-v[i][3]]+p[i][1]+p[i][3]);
			}
			if(v[i][0]>=3&&j>=v[i][1]+v[i][2]+v[i][3]){
				f[j]=max(f[j],f[j-v[i][1]-v[i][2]-v[i][3]]+p[i][1]+p[i][2]+p[i][3]);
			}
		}
		
	}
	cout<<f[m];
	return 0;
}