洛谷P1064 金明的预算方案--动态规则--背包

题目描述

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

主件 附件

电脑 打印机，扫描仪

书柜 图书

书桌 台灯，文具

工作椅 无

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

设第jjj件物品的价格为v[j]v_[j]v[​j]，重要度为w[j]w_[j]w[​j]，共选中了kkk件物品，编号依次为j1,j2,…,jkj_1,j_2,…,j_kj1​,j2​,…,jk​，则所求的总和为：

v[j1]×w[j1]+v[j2]×w[j2]+…+v[jk]×w[jk]v_[j_1] \times w_[j_1]+v_[j_2] \times w_[j_2]+ …+v_[j_k] \times w_[j_k]v[​j1​]×w[​j1​]+v[​j2​]×w[​j2​]+…+v[​jk​]×w[​jk​]。

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

第111行，为两个正整数，用一个空格隔开：

NmN mNm （其中N(<32000)N(<32000)N(<32000)表示总钱数，m(<60)m(<60)m(<60)为希望购买物品的个数。） 从第222行到第m+1m+1m+1行，第jjj行给出了编号为j−1j-1j−1的物品的基本数据，每行有333个非负整数

vpqv p qvpq （其中vvv表示该物品的价格（v<10000v<10000v<10000），p表示该物品的重要度（1−51-51−5），qqq表示该物品是主件还是附件。如果q=0q=0q=0，表示该物品为主件，如果q>0q>0q>0，表示该物品为附件，qqq是所属主件的编号）
输出格式

一个正整数，为不超过总钱数的物品的价格与重要度乘积的总和的最大值（<200000<200000<200000）。
输入输出样例
输入 #1

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

输出 #1

2200

说明/提示

NOIP 2006 提高组 第二题

题解：
考虑一下，如果没有主件和附件的关系约束，这题怎么做，就是很简单的01背包模板问题，对吧，但是现在有了这个主件和附件的约束关系，所以得重新考虑，首先分析最开始01背包模板的这个关系:
dp[j]=max(dp[j],dp[j-v[i]]+v[i]*m[i])；
这个意思是，金钱j减去花费v[i]能得到的价值，选取最大的那个，那么这样考虑，但是如果某个东西是附件，那么再购买的时候，一定得同时买去他的主件，设附件i的主件为sign[i]，那么购买附件时的转移方程就为
dp[j]=max(dp[j],dp[j-v[i]-v[sign[i]]]+v[i]*m[i]+v[sign[i]]*m[sign[i]])；
这个应该很好理解了，因为买附件就必须买主件，但是又有了一个问题，如果在某个状态下，附件i的主件sign[i]已经被购买了呢，就不能用这个状态转移方程了，所以我们要做个判断，判断这种情况下主件是否已经被购买，所以我在这里定义一个动态数组co[maxn],co[j]表示在金额为j时买了哪些东西，然后做判断：
1.如果物件i为主件，直接：
dp[j]=max(dp[j],dp[j-v[i]]+v[i]*m[i])；
2.如果物件i为附件：
2.1如果金额为j-v[i]时已经存在i的主件sign[i]:
dp[j]=max(dp[j],dp[j-v[i]]+v[i]*m[i])；
2.2如果金额为j-v[i]时没有购买主件sign[i]:
dp[j]=max(dp[j],dp[j-v[i]-v[sign[i]]]+v[i]*m[i]+v[sign[i]]*m[sign[i]])；

AC代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=40000;
int dp[maxn],v[maxn],m[maxn],sign[maxn];
vector<int>co[maxn];
int main()
{
	int N,M;
	cin>>N>>M;
	for(int i=1;i<=M;i++)
	cin>>v[i]>>m[i]>>sign[i];
	memset(dp,0,sizeof(dp));
	for(int i=1;i<=M;i++)
	co[i].clear();
	
	for(int i=1;i<=M;i++)
	{
		for(int j=N;j>=v[i];j--)
		{
			if(sign[i]==0)
			{
				int t=dp[j-v[i]]+v[i]*m[i];
				if(t>dp[j])
				{
					dp[j]=t;
					co[j]=co[j-v[i]];
					co[j].push_back(i);
				}
			}
			else
			{
				bool flag=false;
				for(int k=0;k<co[j-v[i]].size();k++)
				if(co[j-v[i]][k]==sign[i])
				{
					flag=true;
					break;
				}
				if(flag)
				{
					int t=dp[j-v[i]]+v[i]*m[i];
					if(t>dp[j])
					{
						dp[j]=t;
						co[j]=co[j-v[i]];
						co[j].push_back(i);
					}
				}
				else
				{
					if(j-v[i]-v[sign[i]]>=0)
					{
						int t=dp[j-v[i]-v[sign[i]]]+v[i]*m[i]+v[sign[i]]*m[sign[i]];
						if(t>dp[j])
						{
							dp[j]=t;
							co[j]=co[j-v[i]-v[sign[i]]];
							co[j].push_back(i);
							co[j].push_back(sign[i]);
						}
					}
				}
			}
		}
	}
	cout<<dp[N]<<endl;
	return 0;
}