leetcode1049 最后一块石头的重量II

题目

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

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

如果 x == y，那么两块石头都会被完全粉碎；
如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。
最后，最多只会剩下一块 石头。返回此石头 最小的可能重量 。如果没有石头剩下，就返回 0。

示例

输入：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]，这就是最优值。

解析

这道题其实也是将物品分成两份，对于本题就是尽量将石头分成重量接近的两堆，这样在相撞后，剩下的石头质量就是最小的了。因此和之前的题目就有一点点相似，用01背包；
然后套用动规五部曲：
1.确定dp数组以及下标的含义
dp[j]表示容量为j的背包，最多可以背最大重量为dp[j]；
2.确定递推公式
dp[j] = max(dp[j], dp[j - stones[i]] + stones[i]); 没什么好说的了，和之前的一样
3.初始化
因为重量都不会是负数，所以dp[j]都初始化为0就可以了，同时dp数组需要初始化为一个较大的空间，一般为target+1，而target=sum/2

func lastStoneWeightII(stones []int) int {
    sum := 0
    for _, value := range stones {
        sum += value
    }
    target := sum / 2
    dp := make([]int, target+1)
    dp[0] = 0 //这一行加不加都行，因为上面默认初始化为0了，但不要自己给初始化成1
    for i := 0; i < len(stones); i++ { //这里也要从0开始1的话部分数据会报错
        for j := target; j >= stones[i]; j-- {
            dp[j] = max(dp[j], dp[j-stones[i]] + stones[i])
        }
    }
    return sum - dp[target] - dp[target]
}

func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}