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

最新推荐文章于 2024-11-12 21:14:07 发布

Charlesjjc

最新推荐文章于 2024-11-12 21:14:07 发布

阅读量403

点赞数

分类专栏： # 动态规划文章标签： java leetcode 动态规划数据结构算法

本文链接：https://blog.csdn.net/qq_37931960/article/details/119793018

版权

动态规划专栏收录该内容

39 篇文章 0 订阅

订阅专栏

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

摘要由CSDN通过智能技术生成

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背包问题，重量一半的情况下，能装下的最大值，使左右两边的石头尽量均衡，这样可以达到碰撞后重量最小。