动态规划_01背包问题
程序员文章站
2022-06-06 17:06:21
...
动态规划来解决问题,三个要素:
**A.最优子结构
B.传递公式
C.边界条件**
例: 在01背包问题中,n=5是物品的数量,c=10是书包能承受的重量,w=[0,2,2,6,5,4]是每个物品的重量,v=[0,6,3,5,4,6]是每个物品的价值,求最大价值。
先把递归的传递公式写出来:
说明:
1.设置w[0] = 0, v[0] = 0
2.当i = 1 时,边界条件为:
if wi>w, c[1,w] = c[0,w] =0
if wi=w, c[1,w] = max(v[1]+c[0,0],c[0,w]) = v[1]
# coding: utf-8
def bag(n,c,w,v):
res=[[0 for j in range(c+1)] for i in range(n+1)]
#~ #错误初始化方法:
#~ res = [[]]
#~ for i in range(n+1):
#~ for j in range(c+1):
#~ res [i][j] = 0
for i in range(1,n+1):
for j in range(1,c+1):
res[i][j]=res[i-1][j]
if j>=w[i] and res[i][j] < res[i-1][j-w[i]] + v[i]:
res[i][j]=res[i-1][j-w[i]]+v[i]
return res
def show(n,c,w,res):
print("Max:",res[n][c])
x=[False for i in range(n+1)]
j=c
for i in range(n,0,-1):
if res[i][j]>res[i-1][j]:
x[i]=True
j-=w[i]
print("Choice: ")
for i in range(n+1):
if x[i]:
print('[',i,']',end='')
print('')
if __name__=='__main__':
n=5
c=10
w=[0,2,2,6,5,4]
v=[0,6,3,5,4,6]
#~ n = 6
#~ c = 10
#~ w=[0,2,2,3,1,5,2]
#~ v=[0,2,3,1,5,4,3]
res=bag(n,c,w,v)
show(n,c,w,res)
下一篇: AC自动机(字符串多模匹配)