开心的金明
【问题描述】
金明今天很开心,家里购置的新房就要领钥匙了,新房里有一间他自己专用的很宽敞的房间。更让他高兴的是,妈妈昨天对他说:“你的房间需要购买哪些物品,怎么布置,你说了算,只要不超过n元钱就行”。今天一早金明就开始做预算,但是他想买的东西太多了,肯定会超过妈妈限定的n元。于是,他把每件物品规定了一个重要度,分为5等:用整数1~5表示,第5等最重要。他还从因特网上查到了每件物品的价格(都是整数元)。他希望在不超过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<30000表示总钱数,m<25为希望购买物品的个数)。
从第2行到第m+1行,第j行给出了编号为j-1的物品的基本数据,每行有2个非负整数v、p,其中v表示该物品的价格(v≤10000),p表示该物品的重要度(1~5)。
【输出格式】
只有一个正整数,为不超过总钱数的物品的价格与重要度乘积的总和的最大值(<100000000)。
【输入样例】
10 5
8 2
4 5
3 5
4 3
2 2
【输出样例】
39
本问题可以简单概括为:有m件物品和一个容量为n的背包。第i件物品的费用是v[i],价格与重要度乘积是p[i]=v[i]×w[i]。求解将哪些物品装入背包可使p[i]总和最大,即“01背包问题”。
1.状态的设计
f[i,j]记录“选取前i件物品花费j元钱”的最大价值,我们可以从第1 个物品出发,一直取到第i个物品,则可以分成i个阶段,我们以f[i,j]作为状态。
钱 | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 |
|
阶段0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 边界 |
阶段1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 16 | 16 | 16 | 取价格8的物品 |
阶段2 | 0 | 0 | 0 | 0 | 20 | 20 | 20 | 20 | 20 | 20 | 20 | 取价格4的物品 |
阶段3 | 0 | 0 | 0 | 15 | 20 | 20 | 20 | 35 | 35 | 35 | 35 | 取价格3的物品 |
阶段4 | 0 | 0 | 0 | 15 | 20 | 20 | 20 | 35 | 35 | 35 | 35 | 取价格4的物品 |
阶段5 | 0 | 0 | 4 | 15 | 20 | 20 | 24 | 35 | 35 | 39 | 39 | 取价格2的物品 |
2.状态转移方程
f[i,j]=max{ f[i-1,j-v[i]]+v[i]×w[i],f[i-1,j]}(1≤i≤m,v[i]≤j≤n)
f[i,0]=0 (1≤i≤m),f[0,j]=0 (0≤j≤n)
该算法的时间复杂度为O(n×m),参考程序如下:
var i,j,m,n:longint; f:array[0..100,0..30000] of longint; v,w:array[1..25] of longint; begin readln(n,m); for i:=1 to m do begin read(v[i],w[i]); for i:=1 to m do for j:=v[i] to n do if f[i-1,j]>f[i-1,j-v[i]]+v[i]*w[i] then f[i,j]:=f[i-1,j] else f[i,j]:=f[i-1,j-v[i]]+v[i]*w[i]; writeln(f[m,n]); end. |
3.优化空间
以上方法的空间复杂度为O(n×m),可以优化到O(n)。
首先,我们知道f[i,j]是由f[i-1,j]和f[i-1,j-v[i]]两个子问题决策而来;也就是说第i阶段只与第i-1阶段有关,可以使用滚动数组来优化空间。
继续分析,若将f[i,j]改为一维数组g[j],并将上面程序中for j:=v[i] to n do循环改为for j:=n downto v[i]do,这样的顺序计算g[j]时,可以保证g[j-v[i]]保存的是前一阶段状态f[i-1,j-v[i]]的值。改进的程序如下:
var i,j,m,n:longint; g:array[0..30000] of longint; v,w:array[1..25] of longint; begin readln(n,m); for i:=1 to m do read(v[i],w[i]); for i:=1 to m do for j:=n downto v[i] do if g[j]<g[j-v[i]]+v[i]*w[i] then g[j]:=g[j-v[i]]+v[i]*w[i]; writeln(g[n]); end. |
请仔细对比两个参考程序,正确理解程序的含义,对今后学习其它类型的背包,有着十分重要的作用。01背包问题是最基本的背包问题,它包含了背包问题中划分阶段,确定状态、建立状态转移方程的基本思想,其它类型的背包问题往往也可以转换成01背包问题求解。