用动态规划算法求解0/1背包问题python完整代码
时间: 2023-09-18 12:13:54 浏览: 148
好的,下面是用动态规划算法求解0/1背包问题的Python完整代码:
```python
def knapsack(weights, values, capacity):
n = len(weights)
dp = [[0] * (capacity + 1) for _ in range(n + 1)]
for i in range(1, n + 1):
for j in range(1, capacity + 1):
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])
return dp[n][capacity]
# 测试
weights = [2, 3, 4, 5]
values = [3, 4, 5, 6]
capacity = 8
print(knapsack(weights, values, capacity)) # 输出:11
```
其中,`weights`和`values`分别是物品的重量和价值的列表,`capacity`是背包的容量。`n`是物品的数量,`dp`是一个`(n+1)×(capacity+1)`的二维列表,`dp[i][j]`表示在前`i`个物品中选择不超过重量为`j`的物品可以得到的最大价值。
我们用两个嵌套的循环来遍历所有的物品和容量。当物品的重量超过当前容量时,我们无法选择这个物品,因此最大价值就是前`i-1`个物品的最大价值,即`dp[i-1][j]`。当物品的重量不超过当前容量时,我们可以选择这个物品,也可以不选择这个物品。如果不选择这个物品,最大价值就是前`i-1`个物品的最大价值,即`dp[i-1][j]`。如果选择这个物品,最大价值就是前`i-1`个物品中不超过重量为`j-weights[i-1]`的物品可以得到的最大价值,加上当前物品的价值,即`dp[i-1][j-weights[i-1]]+values[i-1]`。因此,我们在这两种情况中选择价值更大的方案,更新`dp[i][j]`的值。
最后,返回`dp[n][capacity]`即可得到在不超过背包容量的情况下可以得到的最大价值。
阅读全文