金明的预算(子集型动规)

Description

  金明今天很开心,家里购置的新房就要领钥匙了,新房里有一间金明自己专用的很宽敞的房间。更让他高兴的是,妈妈昨天对他说:“你的房间需要购买哪些物品,怎么布置,你说了算,只要不超过N元钱就行”。今天一早,金明就开始做预算了,他把想买的物品分为两类:主件与附件,附件是从属于某个主件的。
  如果要买归类为附件的物品,必须先买该附件所属的主件。每个主件可以有0个、1个或2个附件。附件不再有从属于自己的附件。金明想买的东西很多,肯定会超过妈妈限定的N元。于是,他把每件物品规定了一个重要度,分为5等:用整数1~5表示,第5等最重要。他还从因特网上查到了每件物品的价格(都是10元的整数倍)。他希望在不超过N元(可以等于N元)的前提下,使每件物品的价格与重要度的乘积的总和最大。
  设第j件物品的价格为v[j],重要度为w[j],共选中了k件物品,编号依次为j1,j2,…,jk,则所求的总和为:v[j1]*w[j1]+v[j2]*w[j2]+ …+v[jk]*w[jk]。(其中*为乘号)
  请你帮助金明设计一个满足要求的购物单。

Input

  第1行,为两个正整数,用一个空格隔开:N m,其中N(<32000)表示总钱数,m(<60)为希望购买物品的个数。  从第2行到第m+1行,第j行给出了编号为j-1的物品的基本数据,每行有3个非负整数:v p q,其中v表示该物品的价格(v<10000),p表示该物品的重要度(1~5),q 表示该物品是主件还是附件。如果 q=0,表示该物品为主件,如果 q>0,表示该物品为附件,q 是所属主件的编号。

Output

  只有一个正整数,为不超过总钱数的物品的价格与重要度乘积的总和的最大值(<200000)。

一,有附件时,可以把主件与任意个附件当做一个整体,也可单独买。二,没有附件也没有从属谁的可以单独买。三,凡是附件都不能单独买。于是相当于前两种情况加起来做01背包问题。这里好想的做法就是一种种讨论,只是代码量略大。

#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXM=65;
const int MAXN=32005;
int N,M,np=0,f[MAXN]={0},v[MAXM],p[MAXM],chi[MAXM][3],belong[MAXM];
int main()
{
	scanf("%d%d",&N,&M);
	int fa;
	for(int i=1;i<=M;i++)
	{
		scanf("%d%d%d",&v[i],&p[i],&fa);
		p[i]*=v[i];
		if(fa)
			chi[fa][++chi[fa][0]]=i,belong[i]=1;
	}
	for(int i=1;i<=M;i++) if(!belong[i])
	for(int j=N;j>=v[i];j--)
	{
		if(chi[i][0]==1)
		{
			int t=chi[i][1],k=0;
			if(j>=v[i])
				k=max(k,f[j-v[i]]+p[i]);
			if(j>=v[i]+v[t])
				k=max(k,f[j-v[i]-v[t]]+p[i]+p[t]);	
			f[j]=max(f[j],k);
		}
		else if(chi[i][0]==2)
		{
			int t1=chi[i][1],t2=chi[i][2],k=0;
			if(j>=v[i])
				k=max(k,f[j-v[i]]+p[i]);
			if(j>=v[i]+v[t1])
				k=max(k,f[j-v[i]-v[t1]]+p[i]+p[t1]);
			if(j>=v[i]+v[t2])
				k=max(k,f[j-v[i]-v[t2]]+p[i]+p[t2]);
			if(j>=v[i]+v[t1]+v[t2])
				k=max(k,f[j-v[i]-v[t1]-v[t2]]+p[i]+p[t1]+p[t2]);
			f[j]=max(f[j],k);
		}
		else if(belong[i]==0)
			f[j]=max(f[j],f[j-v[i]]+p[i]);
	}
	printf("%d",f[N]);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/WWWengine/article/details/81155579