最近涉猎到了动态规划的背包问题,觉得有必要好好学习一下,就写个笔记方便自己之后复习:
背包问题的定义
一般都是给n个物品,它们的重量分别为不同的n个数字,每个物品都有不同的价值,也是不同的n个数字,假设我们有一个背包,装的所有物品的总总量不能超过v,这种约束条件下,求能装的物品的最大价值。
在分析这个问题的时候自己造个数据如下:
weights = [2, 4, 3, 1]
values = [1800, 3000, 2000, 1500]
v = 7
其中weights代表的是4个物品分别的重量;
values代表的是4个物品分别的价值;
v代表的是背包能装的物品的重量的上限;
一、0-1背包问题
每个物品最多只有一个,要么装进去,要么不装进去;
def zero_one_package(weights, values, v):
'''
Args:
weight
value
volumn
Output:
max value
'''
tmp = [0]*v
for i, weight in enumerate(weights):
for weight_idx in range(v, 0, -1):
if weight_idx - weight > 0:
tmp[weight_idx - 1] = max(tmp[weight_idx - 1], tmp[weight_idx - 1 - weight] + values[i])
if weight_idx == weight:
tmp[weight_idx - 1] = max(tmp[weight_idx - 1], values[i])
print(tmp)
return tmp[-1]
print(zero_one_package(weights, values, v))
输出结果为:
[0, 1800, 1800, 1800, 1800, 1800, 1800]
[0, 1800, 1800, 3000, 3000, 4800, 4800]
[0, 1800, 2000, 3000, 3800, 4800, 5000]
[1500, 1800, 3300, 3500, 4500, 5300, 6300]
6300
二、完全背包问题
每个物品有无限个,随便装多少个进书包都没问题,只要总质量满足条件;
def all_package_problem(weights, values, volume):
'''
Args:
weight
value
volumn
Output:
max value
'''
tmp = [0]*volume
for i, weight in enumerate(weights):
for v in range(volume): #实际体积是 v+1
for num in range((v + 1)//weight):
if (v + 1) == weight:
tmp[v] = max(tmp[v], values[i])
else:
tmp[v] = max(tmp[v], tmp[v - (num)*weight] + (num)*values[i])
return tmp[-1]
all_package_problem(weights, values, v)
输出结果为:
[0, 1800, 0, 3600, 1800, 5400, 3600]
[0, 1800, 0, 3600, 1800, 5400, 3600]
[0, 1800, 2000, 3600, 1800, 5400, 5600]
[1500, 3000, 4500, 6000, 7500, 9000, 10500]
10500