小黑算法成长日记12:0-1背包问题(动态规划版)

最新推荐文章于 2023-02-16 15:31:16 发布

爱喝喜茶爱吃烤冷面的小黑黑

最新推荐文章于 2023-02-16 15:31:16 发布

阅读量304

点赞数 1

文章标签：动态规划数据结构算法

本文链接：https://blog.csdn.net/qq_37418807/article/details/106594015

版权

w = [6,3,5,4,6]    # 物品价值
v = [2,2,6,5,4]    # 物品体积
def KnapSack(w,v,c):
    n = len(w)    # 物品个数
    W = w[:]    # 克隆数组，并在0位置插入0，保证元素索引从1开始
    V = v[:]
    W.insert(0,0)
    V.insert(0,0)
    arr = [[0] * (c + 1) for i in range(n + 1)]    # 动态规划数组
    print('动态规划数组如下:')
    for i in range(1,n +  1):
        for j in range(1,c + 1):
            arr[i][j] = arr[i-1][j]
            if j >= V[i] and arr[i][j] < arr[i-1][j-V[i]] + W[i]:    # 判断此物品加入背包是否对子问题最优解有影响
                arr[i][j] = arr[i-1][j-V[i]] + W[i]
            print('{:2d}'.format(arr[i][j]),end = '  ')
        print()
    rest = c
    print('最优解为:')
    for i in range(n,0,-1):
        if arr[i][rest] == arr[i-1][rest]:    # 此物品不在背包内
            continue
        else:
            print(i,end = ' ')    # 此物品在背包内，将此物品拿出
            rest -= V[i]
    print()
    return arr[n][c]
print('最优值为:',KnapSack(w,v,10))