【代码随想录算法训练营】第43天 | 第九章 动态规划（五）+ 复习第14天 第五章 栈与队列（三）

主要内容

动态规划题目

1049. 最后一块石头的重量 II

思路分析

本题其实就是尽量让石头分成重量相同的两堆，相撞之后剩下的石头最小，这样就化解成01背包问题了。

代码

class Solution:
    # 让石头尽可能分成重量相等的两堆，两者差为最后剩下的石头
    def lastStoneWeightII(self, stones: List[int]) -> int:
        s = sum(stones)
        n = len(stones)
        bagweight = s // 2
        dp = [0] * (bagweight + 1)
        for i in range(n): # 遍历物品
            for j in range(bagweight, stones[i] - 1, -1): # 倒序遍历背包
                dp[j] = max(dp[j], dp[j - stones[i]] + stones[i])
        return s - 2 * dp[bagweight]

494. 目标和

思路分析

重点是如何将题目转变为动态规划0-1背包问题
递推公式dp[j] += dp[j-nums[i]],表示装满容量为dp[j]的背包有几种方法

代码

class Solution:
    # 重点在如何转变为0-1背包问题
    # left - right = target, left + right = sum
    # ==> left = (target + sum) / 2
    # 装满容量为left背包有几种方法
    # 递推公式dp[j] += dp[j-nums[i]],表示装满容量为dp[j]的背包有几种方法
    def findTargetSumWays(self, nums: List[int], target: int) -> int:
        n = len(nums)
        s = sum(nums)
        # 无方案
        if (target + s) % 2 == 1 or abs(target) > s:
            return 0
        bagsize = int((target + s) / 2)
        if bagsize < 0:
            return 0
        dp = [0] * (bagsize + 1)
        dp[0] = 1
        for i in range(n): # 遍历物品
            for j in range(bagsize, nums[i] - 1, -1): # 倒序遍历背包
                dp[j] += dp[j - nums[i]]
            # print(dp)
        return dp[bagsize]

474.一和零

思路分析

重点也是如何将问题转换为背包问题
此为2个维度的0-1背包

代码

class Solution:
    # 两个维度的0-1背包问题
    # dp[i][j] 表示最多有i个0和j个1的strs的最大子集的大小为dp[i][j]
    def findMaxForm(self, strs: List[str], m: int, n: int) -> int:
        dp = [[0] * (n + 1) for _ in range(m + 1)]
        for str in strs:
            ones = str.count('1')
            zeros = str.count('0')
            for i in range(m, zeros - 1, -1):
                for j in range(n, ones - 1, -1):
                    dp[i][j] = max(dp[i][j], dp[i - zeros][j - ones] + 1)
        return dp[m][n]