题目在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])