编程语言
首页 > 编程语言> > 多重背包 Ⅰ 0/1背包问题 空间优化 Python

多重背包 Ⅰ 0/1背包问题 空间优化 Python

作者:互联网

题目在AcWing里:4. 多重背包问题 I - AcWing题库 

这题用0/1背包问题的思路可以做,把多个物品逐一枚举。

那么空间优化的逻辑便是:

既然当前物品的体积小于等于枚举的背包体积时状态转移方程是:

ans[ i ][ j ] = max(ans[ i - 1 ][ j ] , value[ i ] + arr[ i - 1 ][ j - value[ i ]]) 

i 表示的是当前行 ,i - 1 表示的是上一行数据

j 表示的是当前列,同时也是枚举到的当前考虑的背包体积。

发现:整个式子只需要考虑两行数据:当前行和上一行,那么原来的思路就会浪费许多不必要的空间(不信自己列图模拟)。

那么再根据取模知识,设取模数为 mod ,取模后的结果只有0 - mod-1之间的循环,把 mod 值定为 2 ,那么便只有0,1,0,1,0,1 ......,如果 i % 2 表示当前行,那么 ( i + 1 ) % 2表示的就会是上一行。 

n,bag = map(int,input().split())
arr = []
all = 0
for _ in range(n):
    a,b,c = map(int,input().split())
    all += c
    while c:
        c-=1
        arr.append([a,b])
new = [[0 for _ in range(bag+1)] for i in range(2)]
for i in range(all):
    get_new = new[(i+1)%2]
    lastNew = new[i%2]
    get_arr = arr[i]
    for j in range(1,bag+1):
        if get_arr[0] > j:get_new[j] = lastNew[j]
        else:get_new[j] = max(lastNew[j],get_arr[1]+lastNew[j-get_arr[0]])
print(new[all%2][-1])

标签:多重,arr,背包,get,Python,range,new,lastNew
来源: https://blog.csdn.net/m0_59010789/article/details/122782240