多重背包 Ⅰ 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