动态规划1049. 最后一块石头的重量 II

前言

之前入门算法用代码随想录（代码随想录），将递归，动态规划这些都学了一遍，也自己动手实现过，但之前也是边参考边写出来的，而且隔了有一段时间了，最近二刷一次！！这次挑战尽量自己写出来。

题目

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

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

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

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

代码

public class _9_1049_最后一块石头的重量 {
    public static void main(String[] args) {
        System.out.println(lastStoneWeightII(new int[]{2, 7, 4, 1, 8, 1})); //1
        System.out.println(lastStoneWeightII(new int[]{31,26,33,21,40})); //5

    }
    public  static int lastStoneWeightII(int[] stones) {
        //求得sum
        int sum = 0;
        for (int stone : stones) {
            sum += stone;
        }

        int target  = sum/2;
        // dp: 在0~i之间能装满容量i的最大价值为dp[i]
        int[] dp = new int[target + 1];

        for (int i = stones[0]; i <= target; i++) {
            dp[i] = stones[0];
        }

        for (int i = 1; i < stones.length; i++) {
            for (int j = target ; j >= 0; j--) {
                if (j - stones[i] >= 0) {
                    dp[j] = Math.max(dp[j], dp[j - stones[i]] + stones[i]);
                }
            }
        }
        return sum - 2 * dp[target];
    }
}