购物单问题C++

购物单

题目描述

如果要买归类为附件的物品，必须先买该附件所属的主件。每个主件可以有 0 个、 1 个或 2 个附件。附件不再有从属于自己的附件。王强想买的东西很多，为了不超出预算，他把每件物品规定了一个重要度，分为 5 等：用整数 1 ~ 5 表示，第 5 等最重要。他还从因特网上查到了每件物品的价格（都是 10 元的整数倍）。他希望在不超过 N 元（可以等于 N 元）的前提下，使每件物品的价格与重要度的乘积的总和最大。
设第 j 件物品的价格为 v[j] ，重要度为 w[j] ，共选中了 k 件物品，编号依次为 j 1 ， j 2 ，……， j k ，则所求的总和为：
v[j 1 ]*w[j 1 ]+v[j 2 ]*w[j 2 ]+ … +v[j k ]*w[j k ] 。（其中 * 为乘号）
请你帮助王强设计一个满足要求的购物单。

思路

本题就是背包问题加了一些限制，可用动态规划求解。

设置三个数组，w[65][3]表示价格，w[i][0]表示第i主件的价格，w[i][1]表示第i个主件的第1个附件的价格，w[i][2]表示第i主件的第2个附件的价格。

v[65][3]表示选中这个主件或附件的总价值，总价值等于价格与重要性的乘积，v[i][0]表示第i主件的价值，v[i][1]表示第i主件的第1个附件的价值，v[i][2]表示第i主件的第2个附件的价值。

d[65][3205]表示余额为3205下前65个物品的最大价值，也就是动态规划的dp数组。

在经过上述整理之后，对于每一个i，都会有四个选择：

• 只买主件i，获得价值v[i][0]
• 买主件+附件1，获得价值v[i][0]+v[i][1]
• 买主件+附件2，获得价值v[i][0]+v[i][2]
• 买主件+附件1+附件2，获得价值v[i][0]+v[i][1]+v[i][2]

分析完成，代码如下：

#include <iostream>
using namespace std;
int w[65][3],v[65][3],d[65][3205];
int main()
{
	int n,m,c,p,q,i,j,t;
	cin>>n>>m;
	n/=10; //都是10的整数倍，因此可以节约空间和时间 
	for(i=1;i<=m;i++)
	{
		cin>>c>>p>>q;
		c/=10; //同上 
		if(q==0) {w[i][q]=c; v[i][q]=c*p;}
		else if(w[q][1]==0) {w[q][1]=c;v[q][1]=c*p;}
		else {w[q][2]=c;v[q][2]=c*p;}
	}
	for(i=1;i<=m;i++)
	for(j=0;j<=n;j++)
	{
		d[i][j]=d[i-1][j];
		if(j>=w[i][0]) {t=d[i-1][j-w[i][0]]+v[i][0];if(t>d[i][j]) d[i][j]=t;}
		if(j>=w[i][0]+w[i][1]) {t=d[i-1][j-w[i][0]-w[i][1]]+v[i][0]+v[i][1];if(t>d[i][j]) d[i][j]=t;}
		if(j>=w[i][0]+w[i][2]) {t=d[i-1][j-w[i][0]-w[i][2]]+v[i][0]+v[i][2];if(t>d[i][j]) d[i][j]=t;}
		if(j>=w[i][0]+w[i][1]+w[i][2]) {t=d[i-1][j-w[i][0]-w[i][1]-w[i][2]]+v[i][0]+v[i][1]+v[i][2];if(t>d[i][j]) d[i][j]=t;}
	}
	cout<<d[m][n]*10<<endl;
	return 0;
}