金明的預算方案(揹包DP)
題目:
金明今天很開心,家裡購置的新房就要領鑰匙了,新房裡有一間金明自己專用的很寬敞的房間。更讓他高興的是,媽媽昨天對他說:“你的房間需要購買哪些物品,怎麼佈置,你說了算,只要不超過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
輸入檔案budget.in 的第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
輸出檔案budget.out只有一個正整數,為不超過總錢數的物品的價格與重要度乘積的總和的最大值(<200000)。
SampleInput
1000 5
800 2 0
400 5 1
300 5 1
400 3 0
500 2 0
SampleOutput
2200
題解:是一個很明顯的揹包問題,只是有附件1和附件2的存在,還有就是價值是重要值乘以價格。
有附件的話就有四種情況。取主件,取主件和附件1,取主件和附件2,取主件和附件12。揹包處理以後花錢最多的不一定就是最優的情況,所以跑一遍迴圈找最大值就是結果。
程式碼:
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<queue>
#include<stack>
#include<cstdio>
#include<cstring>
#include<vector>
#include<cmath>
#include<map>
#define exp 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
using namespace std;
typedef long long LL;
int f[50001];
int a[101],b[101]; //主件的價格和重要值
int a1[101],b1[101];//附件1的價格和重要值
int a2[101],b2[101];//附件2的價格和重要值
int main()
{
int n,m,i,j,ans=0;
scanf("%d%d",&n,&m);
f[0]=0;ans=0;
for(i=1;i<=m;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
if(z==0) //主件
{
a[i]=x;
b[i]=y;
}
else
{
if(b1[z]==0) //附件1
{
a1[z]=x;
b1[z]=y;
}
else //附件2
{
a2[z]=x;
b2[z]=y;
}
}
}
for(i=1;i<=m;i++)
for(j=n;j>=a[i];j--)
{
if (j-a[i]>=0)
f[j]=max(f[j],f[j-a[i]]+a[i]*b[i]);//只買一個主件
if (j-a[i]-a1[i]>=0)
f[j]=max(f[j],f[j-a[i]-a1[i]]+a[i]*b[i]+a1[i]*b1[i]);//買主件和附件一
if (j-a[i]-a2[i]>=0)
f[j]=max(f[j],f[j-a[i]-a2[i]]+a[i]*b[i]+a2[i]*b2[i]);//買主件和附件二
if (j-a[i]-a1[i]-a2[i]>=0)
f[j]=max(f[j],f[j-a[i]-a1[i]-a2[i]]+a[i]*b[i]+a1[i]*b1[i]+a2[i]*b2[i]);//買主件和兩個附件
}
for(i=1;i<=n;i++) //花錢最多的不一定是最優解
if(f[i]>ans) ans=f[i];
printf("%d\n",ans);
return 0;
}