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

最新推荐文章于 2024-11-04 19:24:37 发布

996冲冲冲

最新推荐文章于 2024-11-04 19:24:37 发布

阅读量62

点赞数

分类专栏：动态规划文章标签：算法动态规划

本文链接：https://blog.csdn.net/weixin_44392348/article/details/128937110

版权

动态规划专栏收录该内容

39 篇文章 0 订阅

订阅专栏

该代码实现了一个使用动态规划的方法，来解决选取列表中的石头以使重量最接近列表总重量一半的问题。它通过构建一个二维dp数组，计算从0到i个石头能否组成重量j，最终找到最接近总重量一半的组合，并返回剩余最小的重量差。

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

def lastStoneWeightII(self, stones):
        """
        :type stones: List[int]
        :rtype: int
        """
        #题目目标选则n块石头，使重量最接近总重量的一半
        #最后把题目转换为选n块石头能组成的最靠近总重量一半的数
        #所以定义dp数组为从[0:i]中选一些石头能不能组成重量为j
        #base case:
        #dp[0][0] = 1  dp[i][0] = 1, dp[0][i] = 0 综合一下
        #转移方程：
        #dp[i][j] = dp[i-1][j] or dp[i-1][j-s[i-1]]
        m = len(stones)
        total = sum(stones)
        n = total/2

        dp = [[0]*(n+1) for i in range(m+1)]
        dp[0][0] = 1
        for i in range(1,m+1):
            dp[i][0] = 1
            for j in range(n+1):
                if stones[i-1] > j:
                    dp[i][j] = dp[i-1][j]
                else:
                    dp[i][j] = dp[i-1][j] or dp[i-1][j-stones[i-1]]
        
        for i in range(n,-1,-1):   #找到最接近重量一半的数，再用总重量减去它的两倍就是剩余最小的了
            if dp[m][i]:
                ans = total - 2*i
                return ans
        return 0