题目来源:算法竞赛宝典 第二部 p445
题目
张琪曼等人要为太空战指挥中心购置设备,魔法学院的院长昨天说:“指挥中心需要购买哪些设备,你们研究了算,只要不超过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
思路
刚看题目的时候普普通通的01背包 小事情 然后发现读题不仔细第n次 想了想感觉不好写 去百度了以后看到了纯枚举法 就想着要自己试试把这题用背包的方法写出来
整体思路是从主件的选择入手 选择能加入某主件的情况 再去判断它的附件能不能选择 然后就没有然后了
代码
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
int N,m,f[32005],v[65],p[65],q[65],mark[65];
int main(){
cin>>N>>m;
for(int i=1;i<=m;i++){
cin>>v[i]>>p[i]>>q[i];
}
for(int i=1;i<=m;i++){
if(q[i]==0){
for(int j=N;j>=v[i];j--){
if(f[j]<=f[j-v[i]]+v[i]*p[i]){
f[j]=f[j-v[i]]+v[i]*p[i];
mark[i]=1;
if(mark[i]){
for(int k=1;k<=m;k++){
if(mark[q[k]]==1)
for(int l=N;l>=v[k]+f[j];l--){
f[l]=max(f[l],f[l-v[k]]+v[k]*p[k]);
}
}
}
mark[i]=0;
}
}
}
}
cout<<f[N];
return 0;
}