01背包问题求解

来源于 https://kamacoder.com/problempage.php?pid=1046

使用动态规划,五步走

1.定义状态数组和具体状态含义:

dp是个二维数组,第一维代表物品索引,第二维代表背包空间状态。

dp[i][j]是指物品i 在背包空间j 的情况下所能放的最大价值。

2.明确状态转移公式

根据题意,要么我放物品i ,要么不放。由易到难:

先考虑不放,那一定是我背包空间不够这个物品的体积才不放。所以此时dp[i][j] 的状态转移就是继承在背包空间j 的时候,物品i-1 (也就是上一个物品)时刻的状态: dp[i][j] = dp[i-1][j]

再考虑放 ,放的话首先要加上物品i 的价值,然后继承上一个能和物品i 的体积适配(也就是背包空间j去掉物品i 的体积后所剩的背包空间,在这个背包空间j-weigths[i-1]下的前一个物品i-1所得到的最大价值)的状态: dp[i][j] = dp[i-1][j-weigths[i-1]] + values[i-1]

这里为什么不是继承前一个物品在空间j下的状态,因为这样很可能两个物品加起来背包塞不下,为了尝试在物品i 时能成功塞到空间j 里,必须继承背包空间j-weigths[i-1]时的状态。

综合下,能够放的情况下,我也可以不放,这样就省了空间,有可能在后面有更有价值的。

所以考虑放的情况下也是放与不放取最大值。

问:为什么明明是物品i,空间j,但是每次都是weigths[i-1]、values[i-1]?

答:这与dp初始化有关,我在初始化时考虑了物品0和空间0,都代表不存在。但是weigths是列表,访问物品1的时候索引是0,物品0是不存在的。
问:什么是背包空间不够这个物品的体积?

答:如果空间j < weights[i-1],说明物品i 的体积超过了当前空间。

3.初始化边界条件

根据定义,物品i为0时,代表没有物品,所以dp[0][j]=0

空间j为0时,代表没有空间,放不进去,所以dp[i][j]=0

4.遍历方式确定

按照先遍历一维,再遍历二维的方式。先遍历每个物品,再遍历每个背包空间。

5.举例推导dp

代码

ACM模式

python 复制代码
import sys

# 从标准输入读取数据
input = sys.stdin.read
data = input().split()

# 解析第一个整数m(研究材料的种类)和第二个整数n(小明的行李空间)
m, n = int(data[0]), int(data[1])

# 解析研究材料所占空间的列表weights和研究材料价值的列表values
weights = list(map(int, data[2:2 + m]))
values = list(map(int, data[2 + m:2 + 2 * m]))

# 动态规划
# 状态数组dp,dp[i][j]表示前i个物品放入容量为j的背包中可以获得的最大价值
dp = [[0 for _ in range(n + 1)] for _ in range(m + 1)]

# 填充状态数组dp
for i in range(1, m + 1):  # 遍历每一个物品
    for j in range(1, n + 1):  # 遍历背包容量从1到n
        if weights[i - 1] > j:  # 如果当前物品的重量大于当前背包容量
            dp[i][j] = dp[i - 1][j]  # 当前物品不能放入背包,继承上一个状态
        else:
            # 当前物品可以放入背包,选择放入或不放入,取最大值
            dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weights[i - 1]] + values[i - 1])

# 输出dp数组中dp[m][n]的值,即前m个物品放入容量为n的背包中可以获得的最大价值
print(dp[m][n])

      

时间复杂度:O(m * n),因为需要遍历每个物品和每个背包容量。

空间复杂度:O(m * n),因为需要一个二维数组来存储中间结果。

相关推荐
AI极客菌1 小时前
Controlnet作者新作IC-light V2:基于FLUX训练,支持处理风格化图像,细节远高于SD1.5。
人工智能·计算机视觉·ai作画·stable diffusion·aigc·flux·人工智能作画
阿_旭1 小时前
一文读懂| 自注意力与交叉注意力机制在计算机视觉中作用与基本原理
人工智能·深度学习·计算机视觉·cross-attention·self-attention
王哈哈^_^1 小时前
【数据集】【YOLO】【目标检测】交通事故识别数据集 8939 张,YOLO道路事故目标检测实战训练教程!
前端·人工智能·深度学习·yolo·目标检测·计算机视觉·pyqt
Power20246662 小时前
NLP论文速读|LongReward:基于AI反馈来提升长上下文大语言模型
人工智能·深度学习·机器学习·自然语言处理·nlp
数据猎手小k2 小时前
AIDOVECL数据集:包含超过15000张AI生成的车辆图像数据集,目的解决旨在解决眼水平分类和定位问题。
人工智能·分类·数据挖掘
好奇龙猫2 小时前
【学习AI-相关路程-mnist手写数字分类-win-硬件:windows-自我学习AI-实验步骤-全连接神经网络(BPnetwork)-操作流程(3) 】
人工智能·算法
沉下心来学鲁班2 小时前
复现LLM:带你从零认识语言模型
人工智能·语言模型
数据猎手小k2 小时前
AndroidLab:一个系统化的Android代理框架,包含操作环境和可复现的基准测试,支持大型语言模型和多模态模型。
android·人工智能·机器学习·语言模型
YRr YRr2 小时前
深度学习:循环神经网络(RNN)详解
人工智能·rnn·深度学习
sp_fyf_20243 小时前
计算机前沿技术-人工智能算法-大语言模型-最新研究进展-2024-11-01
人工智能·深度学习·神经网络·算法·机器学习·语言模型·数据挖掘