有一堆石头,用整数数组 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