代码随想录算法训练营第43天 | 1049.最后一块石头的重量 II + 494.目标和 + 474.一和零

今日任务

目录

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

494.目标和 - Medium

474.一和零 - Medium

---

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

题目链接：力扣-1049. 最后一块石头的重量 II

有一堆石头，用整数数组 stones 表示。其中 stones[i] 表示第 i 块石头的重量。

每一回合，从中选出任意两块石头，然后将它们一起粉碎。假设石头的重量分别为 x 和 y，且 x <= y。那么粉碎的可能结果如下：

• 如果 x == y，那么两块石头都会被完全粉碎；
• 如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。

最后，最多只会剩下一块 石头。返回此石头 最小的可能重量 。如果没有石头剩下，就返回 0。

提示：尽量让石头分成重量相同的两堆，相撞之后剩下的石头最小

class Solution:
    def lastStoneWeightII(self, stones: List[int]) -> int:
        _sum = sum(stones)
        bag = _sum//2

        dp = [0] * (bag + 1)
        for i in range(len(stones)):
            for j in range(bag, stones[i]-1, -1):
                dp[j] = max(dp[j], dp[j-stones[i]]+stones[i])

        return _sum - dp[bag] - dp[bag]

494.目标和 - Medium

题目链接：力扣-494. 目标和

给你一个整数数组 nums 和一个整数 target 。

向数组中的每个整数前添加 '+' 或 '-' ，然后串联起所有整数，可以构造一个 表达式 ：

• 例如，nums = [2, 1] ，可以在 2 之前添加 '+' ，在 1 之前添加 '-' ，然后串联起来得到表达式 "+2-1" 。

返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。

提示：left组合 - right组合 = target；left + right = sum，推导出 left = (target + sum)/2；此时问题就是在集合nums中找出和为left的组合

class Solution:
    def findTargetSumWays(self, nums: List[int], target: int) -> int:
        _sum = sum(nums)
        if abs(target) > _sum:
            return 0
        if (_sum + target) % 2 == 1:
            return 0
        bigSize = (_sum + target) // 2
        dp = [0] * (bigSize + 1)
        dp[0] = 1
        for num in nums:
            for j in range(bigSize, num-1, -1):
                dp[j] += dp[j-num]
        return dp[bigSize]

474.一和零 - Medium

题目链接：力扣-474. 一和零

给你一个二进制字符串数组 strs 和两个整数 m 和 n 。

请你找出并返回 strs 的最大子集的长度，该子集中 最多 有 m 个 0 和 n 个 1 。

如果 x 的所有元素也是 y 的元素，集合 x 是集合 y 的 子集 。

提示：01背包问题，只不过这个背包有两个维度，一个是m 一个是n，而不同长度的字符串就是不同大小的待装物品; dp[i][j]表示最多有i个0和j个1的strs的最大子集的大小

class Solution:
    def findMaxForm(self, strs: List[str], m: int, n: int) -> int:
        dp = [[0] * (n+1) for _ in range(m+1)]
        for s in strs:
            zeroNum = s.count('0')
            oneNum = len(s) - zeroNum
            for i in range(m, zeroNum-1, -1):
                for j in range(n, oneNum-1, -1):
                    dp[i][j] = max(dp[i][j], dp[i-zeroNum][j-oneNum]+1)
        return dp[m][n]