代码随想录算法训练营Day41 | 01背包问题（二维）、01背包问题（一维）、416.分割等和子集 | Python | 个人记录向

01背包问题（二维）

代码随想录：01背包问题（二维）
卡码网：01背包问题（二维）

做题

之前做过，但需要回顾一下思路。

看文章

大概翻了一下文章，然后就写代码。关键点是：dp[i][j] 表示从下标为[0-i]的物品里任意取，放进容量为j的背包，价值总和最大是多少。

m, n = map(int, input().split())
weight = list(map(int, input().split()))
value = list(map(int, input().split()))

dp = [[0] * (n+1) for _ in range(m)]
if weight[0] <= n:
    for i in range(weight[0], n+1):
        dp[0][i] = value[0]

for i in range(1, m):
    for j in range(n+1):
        if j >= weight[i]:
            dp[i][j] = max(dp[i-1][j], dp[i-1][j-weight[i]]+value[i])
        else:
            dp[i][j] = dp[i-1][j]

print(dp[m-1][n])

01背包问题（一维）

代码随想录：01背包问题（一维）
Leetcode：01背包问题（一维）

做题

同样需要回顾一下思路。

看文章

主要是用滚动数组代替二维数组，但区别在于，要倒序遍历，从而避免在一轮遍历中使用了该轮遍历修改过的数值。

m, n = map(int, input().split())
weight = list(map(int, input().split()))
value = list(map(int, input().split()))

dp = [0] * (n+1)

for i in range(m):
    for j in range(n, -1, -1):
        if j >= weight[i]:
            dp[j] = max(dp[j], dp[j-weight[i]]+value[i])

print(dp[n])

416.分割等和子集

代码随想录：416.分割等和子集
Leetcode：416.分割等和子集

做题

无思路。

看文章

将该问题视为01背包问题，当刚好能装下sum / 2的数字时，就说明可以分割。

class Solution:
    def canPartition(self, nums: List[int]) -> bool:
        nums_sum = sum(nums)
        if sum(nums) % 2 != 0:
            return False
        package = nums_sum // 2

        dp = [0] * (package+1)
        for i in range(len(nums)):
            for j in range(package, -1, -1):
                if j >= nums[i]:
                    dp[j] = max(dp[j], dp[j-nums[i]] + nums[i])
        
        if dp[package] == package:
            return True
        else:
            return False

时间复杂度：O(n^2)
空间复杂度：O(n)

以往忽略的知识点小结

• 01背包问题
• 滚动数组
• 将其他问题转化为01背包问题

个人体会

完成时间：2h30min。
心得：01背包问题、滚动数组还是挺有意思的，将其他问题转化成背包问题的思路更有意思，需要多理解并掌握。

​