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

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

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

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

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

示例 1:

输入:stones = [2,7,4,1,8,1]
输出:1
解释:
组合 2 和 4,得到 2,所以数组转化为 [2,7,1,8,1],
组合 7 和 8,得到 1,所以数组转化为 [2,1,1,1],
组合 2 和 1,得到 1,所以数组转化为 [1,1,1],
组合 1 和 1,得到 0,所以数组转化为 [1],这就是最优值。

示例 2:

输入:stones = [31,26,33,21,40]
输出:5

思路:这道题和1046那道题的区别就是那道题明说了每次都取最大的两块,因此结果是定死的,但这道题每次是任意取2块,不同的取法得到的答案也不同,要我们返回最小的结果,根据例子,我们也无法找出一种明确的取法,每次是取怎样的两块。

因此这道题该怎么做呢,其实我们不应该拘泥于每次取哪两块,而是整体去考虑,每次都是两块两块的取,我们可以看成是从两堆石头中取,也就是把原来的一堆石头分成两堆,每次都从各自的一堆中取一个,那怎么分成两堆,能使得最后得到的石头最小呢,一定是这两堆的总重量尽可能相近,这样相撞之后得到的石头会是最小,最后得到的石头重量就是两堆的重量相减。

Ok,这就把问题转化成0-1背包问题了,x = sumNum / 2,也就是其中一堆的重量,我们计算重量为x的这个背包(堆)最大能放多少重量的石头。另一堆石头的重量就是sumNum-dp[x],因此最后剩下的石头重量为sumNum-dp[x]-dp[x]。

代码(Python):

class Solution(object):
    def lastStoneWeightII(self, stones):
        if len(stones) == 1: 
            return stones[0]
        sumNum = sum(stones)
        x = sumNum / 2
        dp = [0] * (x+1)
        dp[0] = 0
        for i in range(0,len(stones)):
            for j in range(x,stones[i]-1,-1):
                dp[j] = max(dp[j],dp[j-stones[i]]+stones[i])
        return sumNum - dp[x]*2                              
  • 7
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值