【NOIP2006提高组】金明的预算

【问题描述】

	金明今天很开心，家里购置的新房就要领钥匙了，新房里有一间金明自己专用的很宽敞的房间。更让他高兴的是，妈妈昨天对他说：“你的房间需要购买哪些物品，怎么布置，你说了算，只要不超过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]。（其中*为乘号） 　　请你帮助金明设计一个满足要求的购物单。

	【输入格式】
		第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 是所属主件的编号。

	【输出格式】
		只有一个正整数，为不超过总钱数的物品的价格与重要度乘积的总和的最大值（<200000）。

	【输入样例】
		1000 5 800 2 0 400 5 1 300 5 1 400 3 0 500 2 0

	【输出样例】
		2200

题目大意:给出一些物品,这些物品可能是主件或者附件,现在要求从这些物品中选择一些,使得所有物品的总价格小于等于n的前提下使得每件物品的价格与重要度的乘积的总和最大。

乍一看是普通的0/1背包问题，但是题目中的物品都有主件附件之分，而且要购买一件附件之前必须购买对应的主件，又因为每个主件可以有0、1、2个附件。

所以对于不同的主件及其附件，不同的选择有:

如果没有附件:那只能不选或选主件。如果有一个附件，除了不选和选主件之外还可以选1个主件和附件。如果有两个附件，除了上述选择之外还可以选主件和第二件附件，或者都选。

在读入数据的时候就将主件和相关的附件压到一个数组中，然后根据元素个数进行不同的选择，对各种选择取最大值即可。

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<set>
#define maxn 65
using namespace std;
int n,m;
int f[35000];
struct data
{
	int v,p,q;	
}A[maxn];
vector<data>g[maxn];

bool cmp(data a,data b)
{
	return a.q<b.q;
}
/*
	f(i,j)=在前i件物品选择一些,在总价格j元内所能达到的最大重要度和价格乘积和
	f(i,j)=max(f(i-1,j),f(只选主件),(f(选主件和其中一件附件)(有可能有两种)),f(都选))
	f(0,0)=0;
*/
int main()
{
	//freopen("my.in","r",stdin);
	//freopen("my.out","w",stdout);
	
	scanf("%d%d",&n,&m);
	
	//memset(cnt,0,sizeof(cnt));
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&A[i].v,&A[i].p,&A[i].q);
		if(A[i].q==0)
		{
			g[i].push_back(A[i]);
		}
		else
		{
			g[A[i].q].push_back(A[i]);
		}
	}
	
	for(int i=1;i<=m;i++)if(g[i].size()>0) 
	sort(g[i].begin(),g[i].end(),cmp);
	
	
	for(int i=1;i<=m;i++)
	for(int j=n;j>=0;j--)
	{
		int t1=f[j],t2=0,t3=0,t4=0,t5=0;
		if(g[i].size()==1)
		{
			if(j-A[i].v>=0)t2=f[j-A[i].v]+A[i].v*A[i].p;
		}
		else if(g[i].size()==2)
		{
			if(j-A[i].v>=0)t2=f[j-A[i].v]+A[i].v*A[i].p;
			if(j-A[i].v-g[i][1].v>=0)t3=f[j-A[i].v-g[i][1].v]+A[i].v*A[i].p+g[i][1].v*g[i][1].p;
		}
		else if(g[i].size()==3)
		{
			if(j-A[i].v>=0)t2=f[j-A[i].v]+A[i].v*A[i].p;
			if(j-A[i].v-g[i][1].v>=0)t3=f[j-A[i].v-g[i][1].v]+A[i].v*A[i].p+g[i][1].v*g[i][1].p;
			if(j-A[i].v-g[i][2].v>=0)t4=f[j-A[i].v-g[i][2].v]+A[i].v*A[i].p+g[i][2].v*g[i][2].p;
			if(j-A[i].v-g[i][1].v-g[i][2].v>=0)t5=f[j-A[i].v-g[i][1].v-g[i][2].v]+A[i].v*A[i].p+g[i][1].v*g[i][1].p+g[i][2].v*g[i][2].p;
		}
		
		f[j]=max(t1,max(t2,max(t3,max(t4,t5)))); 
		
	}
	
	printf("%d",f[n]);
	return 0;
}