代码随想录算法训练营第四十二天 |01背包问题(二维)、01背包问题(一维滚动数组)、416. 分割等和子集

134 阅读6分钟

01背包问题(二维)

代码随想录视频讲解

代码随想录文章讲解

有n件物品和一个最多能背重量为w 的背包。第i件物品的重量是weight[i],得到的价值是value[i] 。每件物品只能用一次,求解将哪些物品装入背包里物品价值总和最大。

动态规划

  1. 确定dp数组(dp table)以及下标的含义

    对于背包问题,有一种写法, 是使用二维数组,即**dp[i][j]表示从下标为[0-i]的物品里任意取,放进容量为j的背包,价值总和最大是多少**。

  2. 确定递推公式

    再回顾一下dp[i][j]的含义:从下标为[0-i]的物品里任意取,放进容量为j的背包,价值总和最大是多少。

    那么可以有两个方向推出来dp[i][j]

    • 不放物品i:由dp[i - 1][j]推出,即背包容量为j,里面不放物品i的最大价值,此时dp[i][j]就是dp[i - 1][j]。(其实就是当物品i的重量大于背包j的重量时,物品i无法放进背包中,所以被背包内的价值依然和前面相同。)
    • 放物品i:由dp[i - 1][j - weight[i]]推出,dp[i - 1][j - weight[i]]为背包容量为j - weight[i]的时候不放物品i的最大价值,那么dp[i - 1][j - weight[i]] + value[i](物品i的价值),就是背包放物品i得到的最大价值

    所以递归公式:dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);

  3. dp数组的初始化

    • 首先从dp[i][j]的定义出发,如果背包容量j0的话,即dp[i][0],无论是选取哪些物品,背包价值总和一定为0

    • 状态转移方程 dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);可以看出i是由i-1 推导出来,那么i0的时候就一定要初始化。

    • dp[0][j],即:i0,存放编号0的物品的时候,各个容量的背包所能存放的最大价值。那么很明显当j < weight[0]的时候,dp[0][j]应该是 0,因为背包容量比编号0的物品重量还小。当j >= weight[0]时,dp[0][j]应该是value[0],因为背包容量放足够放编号0物品。

      for (int j = 0 ; j < weight[0]; j++) {  
        // 当然这一步,如果把dp数组预先初始化为0了,这一步就可以省略,但很多同学应该没有想清楚这一点。
          dp[0][j] = 0;
      }
      // 正序遍历
      for (int j = weight[0]; j <= bagweight; j++) {
          dp[0][j] = value[0];
      }
      
    • dp[0][j]dp[i][0]都已经初始化了,那么其他下标应该初始化多少呢?其实从递归公式:dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);可以看出dp[i][j]是由左上方数值推导出来了,那么 其他下标初始为什么数值都可以,因为都会被覆盖。初始-1,初始-2,初始100,都可以!但只不过一开始就统一把dp数组统一初始为0,更方便一些。

  4. 确定遍历顺序

    • 那么问题来了,先遍历 物品还是先遍历背包重量呢?* 其实都可以!! 但是先遍历物品更好理解*。
    • 虽然两个for循环遍历的次序不同,但是dp[i][j]所需要的数据就是左上角,根本不影响dp[i][j]公式的推导!
    def knapsack_problem_2d(bag_size, weights, values):
        rows, cols = len(values), bag_size + 1
        dp = [[0] * cols for _ in range(rows)]
    ​
        # initialize row 0
        for j in range(weights[0], cols):
            dp[0][j] = values[0]
    ​
        for i in range(1, rows):
            cur_weight, cur_val = weights[i], values[i]
            for j in range(1, cols):
                if cur_weight > j:
                    dp[i][j] = dp[i - 1][j]
                else:
                    dp[i][j] = max(dp[i-1][j], dp[i-1][j - cur_weight] + cur_val)
    ​
        return dp[-1][-1]
    ​
    ​
    if __name__ == "__main__":
        bag_size = 4
        weight = [1, 3, 4]
        value = [15, 20, 30]
        knapsack_problem_2d(bag_size, weight, value)
    

01背包问题(一维滚动数组)

代码随想录视频讲解

代码随想录文章讲解

动态规划

  1. 确定dp数组(dp table)以及下标的含义

    在一维dp数组中,dp[j]表示:容量为j的背包,所背的物品价值可以最大为dp[j]

  2. 确定递推公式

    dp[j]为 容量为j的背包所背的最大价值,那么如何推导dp[j]呢?

    dp[j]可以通过dp[j - weight[i]]推导出来,dp[j - weight[i]]表示容量为j - weight[i]的背包所背的最大价值。

    dp[j - weight[i]] + value[i]表示 容量为 j - 物品i重量 的背包 加上 物品i的价值。(也就是容量为j的背包,放入物品i了之后的价值即:dp[j]

    此时dp[j]有两个选择,一个是取自己dp[j]相当于 二维dp数组中的dp[i-1][j],即不放物品i,一个是取dp[j - weight[i]] + value[i],即放物品i,指定是取最大的,毕竟是求最大价值,

    所以递归公式为:

    dp[j] = max(dp[j], dp[j - weight[i]] + value[i]);
    

    可以看出相对于二维dp数组的写法,就是把dp[i][j]i的维度去掉了。

  3. dp数组的初始化

    dp[j]表示:容量为j的背包,所背的物品价值可以最大为dp[j],那么dp[0]就应该是0,因为背包容量为0所背的物品的最大价值就是0

    那么dp数组除了下标0的位置,初始为0,其他下标应该初始化多少呢?

    看一下递归公式:dp[j] = max(dp[j], dp[j - weight[i]] + value[i]);

    dp数组在推导的时候一定是取价值最大的数,如果题目给的价值都是正整数那么非0下标都初始化为0就可以了。

    这样才能让dp数组在递归公式的过程中取的最大的价值,而不是被初始值覆盖了

    那么我假设物品价值都是大于0的,所以dp数组初始化的时候,都初始为0就可以了。

  4. 确定遍历顺序

    • 一维dp遍历的时候,背包是从大到小。倒序遍历是为了保证物品i只被放入一次! 。但如果一旦正序遍历了,那么物品0就会被重复加入多次!
    • 再来看看两个嵌套for循环的顺序,代码中是先遍历物品嵌套遍历背包容量,那可不可以先遍历背包容量嵌套遍历物品呢? 不可以!
    def knapsack_problem_1d(bag_size, weights, values):
        dp = [0] * (bag_size + 1)
    ​
        for i in range(len(values)):
            cur_weight, cur_val = weights[i], values[i]
            for j in range(bag_size, cur_weight - 1, -1):
                dp[j] = max(dp[j], dp[j - cur_weight] + cur_val)
    ​
        return dp[-1]
    ​
    ​
    if __name__ == "__main__":
        bag_size = 4
        weight = [1, 3, 4]
        value = [15, 20, 30]
        print(knapsack_problem_1d(bag_size, weight, value))
    

416. 分割等和子集

代码随想录文章讲解

动态规划

  • 01背包问题:

    • 背包容量为sum/2
    • 背包要放入的商品(集合里的元素)重量为 元素的数值,价值也为元素的数值
    • 背包如果正好装满,说明找到了总和为 sum / 2 的子集。
    • 背包中每一个元素是不可重复放入。
class Solution:
    def canPartition(self, nums: List[int]) -> bool:
        target = sum(nums)
        if target % 2 == 1: 
            return False
        partition_sum = target // 2
        
        dp = [0] * (partition_sum + 1)
        
        for i in range(len(nums)):
            for j in range(partition_sum, nums[i] - 1, -1):
                dp[j] = max(dp[j], dp[j - nums[i]] + nums[i])
        
        return dp[partition_sum] == partition_sum