算法训练 Day42

最新推荐文章于 2024-10-15 14:48:32 发布

m0_73129924

最新推荐文章于 2024-10-15 14:48:32 发布

阅读量86

点赞数

文章标签：算法 leetcode 职场和发展 python

本文链接：https://blog.csdn.net/m0_73129924/article/details/129220507

版权

文章提供了三个使用动态规划解决的问题：1049.最后一块石头的重量II，通过寻找最优分割使两堆石头的重量差最小；494.目标和，寻找数组中元素加减得到目标和的方法数；474.一和零，求在限制0和1数量的情况下，能选取的字符串最大数量。每个问题都涉及背包问题的变种，通过二维或一维动态规划数组进行求解。

摘要由CSDN通过智能技术生成

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

解题思路：动规

class Solution:
    def lastStoneWeightII(self, stones: List[int]) -> int:
        total = sum(stones)
        #将石头分成两堆差不多大小的，尽可能大得取到其中一堆，这样可以有尽可能小的重量返回
        target = total // 2
        dp = [0] *(target+1)
        for i in range(len(stones)):
            for j in range(target,stones[i]-1,-1):
                print(dp)
                dp[j] = max(dp[j],dp[j-stones[i]] + stones[i])
        return (total - dp[target]) - dp[target]

494. 目标和

class Solution:
    def findTargetSumWays(self, nums: List[int], target: int) -> int:
        #dp[j] 装满容量为j的背包有dp[j]种方法
        # nums:物品重量
        total = sum(nums) 
        if abs(target) > total or (total + target) % 2 != 0: return 0
        # 背包容量
        total += target
        bag_size = total // 2

        dp = [0] * (bag_size+1)
        dp[0] = 1
        for i in range(len(nums)):
            for j in range(bag_size,nums[i]-1,-1):
                # print(dp)
                dp[j] += dp[j-nums[i]]
        return dp[-1]

474.一和零

class Solution:
    def findMaxForm(self, strs: List[str], m: int, n: int) -> int:
        #dp[i][j]：装满i个0，j个1最多能装dp[i][j]个物品
        #背包重量有两个维度，m个0 和 n个1
        #物品重量：x个0，y个1

        #初始化
        dp = [[0]*(n+1) for _ in range(m+1)]

        #物品遍历
        for num in strs:
            x = num.count('0')
            y = num.count('1')
            #背包二维遍历
            for i in range(m,x-1,-1):
                for j in range(n,y-1,-1):
                    dp[i][j] = max(dp[i][j],dp[i-x][j-y]+1)
        return dp[m][n]