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

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

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

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

设第 𝑗j 件物品的价格为 𝑣𝑗vj​，重要度为𝑤𝑗wj​，共选中了 𝑘k 件物品，编号依次为 𝑗1,𝑗2,…,𝑗𝑘j1​,j2​,…,jk​，则所求的总和为：

vj1​​×wj1​​+vj2​​×wj2​​+⋯+vjk​​×wjk​​。

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

输入格式

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

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

输出格式

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

输入输出样例

输入 #1

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

输出 #1

2200

说明/提示

数据规模与约定

对于全部的测试点，保证 1≤n≤3.2×10^4，1≤m≤60，0≤vi​≤10^4，1≤pi​≤5，0≤qi​≤m，答案不超过2×10^5。

思路：

将每个主件的附件存储到主件上去，只对主件进行dp，由于每个主件最多有两个附件，那么就会有5种情况：不选主件、只选主件、主件加附件1、主件加附件2、主件加附件1和2。

对于每种情况进行一次dp

状态转移方程：

dp[i,j]=max(dp[i,j] , dp[i-1][j-添加的价值]+sum(添加的重要度i*添加的价值i))

代码：

#include<bits/stdc++.h>
using namespace std;
int w[100],p[100],d;
int dp[65][340010];
vector<int>v[65];
int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&p[i],&w[i],&d);
		v[d].push_back(i);
	}
	for(int i=1;i<=v[0].size();i++)
	{
		int f=v[0][i-1];
		for(int j=n;j>=0;j--)
		{
			dp[i][j]=max(dp[i][j],dp[i-1][j]);
			if(j-p[f]>=0){
				dp[i][j]=max(dp[i][j],dp[i-1][j-p[f]]+w[f]*p[f]);
			}
			if(v[f].size()>=1&&j-p[f]-p[v[f][0]]>=0){
				dp[i][j]=max(dp[i][j],dp[i-1][j-p[f]-p[v[f][0]]]+w[f]*p[f]+w[v[f][0]]*p[v[f][0]]);
			}
			if(v[f].size()>=2&&j-p[f]-p[v[f][0]]-p[v[f][1]]>=0){
				dp[i][j]=max(dp[i][j],dp[i-1][j-p[f]-p[v[f][0]]-p[v[f][1]]]+w[f]*p[f]+w[v[f][0]]*p[v[f][0]]+w[v[f][1]]*p[v[f][1]]);
			}
			if(v[f].size()>=2&&j-p[f]-p[v[f][1]]>=0){
				dp[i][j]=max(dp[i][j],dp[i-1][j-p[f]-p[v[f][1]]]+w[f]*p[f]+w[v[f][1]]*p[v[f][1]]);
			}
		}
	}
	cout<<dp[v[0].size()][n];
	return 0;
}