ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

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

2022-02-04 12:33:17  阅读:165  来源: 互联网

标签:多重 arr 背包 get Python range new lastNew


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

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有