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

最新推荐文章于 2025-12-02 21:20:38 发布

原创最新推荐文章于 2025-12-02 21:20:38 发布 · 458 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #leetcode #动态规划 #数据结构 #算法

动态规划专栏收录该内容

39 篇文章

订阅专栏

这篇博客介绍了如何将问题转化为0-1背包问题，通过动态规划求解在给定石头重量列表中找到使得两部分石头尽可能均衡的方案，以达到最后一块石头重量最小的目标。算法涉及到了数组处理、动态规划状态转移和最大值计算。

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

class Solution {
    public int lastStoneWeightII(int[] stones) {
        int sum = 0;

        for(int i = 0;i < stones.length;i++){
            sum += stones[i];
        }
        int target = sum / 2;
        int[][] dp = new int[stones.length + 1][target + 1];

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

        int res = sum - 2*dp[stones.length][target];
        return res;

    }
}

class Solution {
    public int lastStoneWeightII(int[] stones) {
        int sum = 0;

        for(int i = 0;i < stones.length;i++){
            sum += stones[i];
        }
        int target = sum / 2;
        int[] dp = new int[target + 1];

        for(int i = 1; i <= stones.length;i++){
            int x = stones[i - 1];
            for(int j = target; j >= x;j--){
                dp[j] = Math.max(dp[j],dp[j - x] + x);
            }
        }

        int res = sum - 2*dp[target];
        return res;

    }
}

思路还是可以转变为0-1背包问题，重量一半的情况下，能装下的最大值，使左右两边的石头尽量均衡，这样可以达到碰撞后重量最小。