最后一块石头的重量 II java

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

输入:stones = [1,2]
输出:1

提示:

1 <= stones.length <= 30
1 <= stones[i] <= 100

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/last-stone-weight-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:一开始想用贪心的(最大的撞第二大的),思考了一下感觉可能是错的(果然是错的)。然后想了一下动态规划咋做,想到了昨天的有点类似,就自己写了一下,结果就比官方的的多3ms。思路如下:
找两堆石头,让他们的差尽量小。题目又让我们尽可能全部粉碎,所以没堆石头sum/2是最完美的。通过这个思路,就变成了容量为sum/2的01背包问题。

class Solution {
    public int lastStoneWeightII(int[] stones) {
        int sum = 0;
        int n = stones.length;
        for(int i=0;i<n;i++){
            sum += stones[i];
        }
        int[][] dp = new int[stones.length+1][sum/2+1];
        for(int i=1;i<dp.length;i++)
            for(int j=1;j<dp[0].length;j++)
            {
                dp[i][j] = Math.max(dp[i][j-1],dp[i-1][j]);
                if(stones[i-1]<=j){
                    dp[i][j] = Math.max(dp[i][j],dp[i-1][j-stones[i-1]]+stones[i-1]);
                }
            }
        
        int ans = sum - dp[stones.length][sum/2]*2;
        return ans;
    }
}
// class Solution {
//     public int lastStoneWeightII(int[] stones) {
//         int sum = 0;
//         for (int weight : stones) {
//             sum += weight;
//         }
//         int n = stones.length, m = sum / 2;
//         boolean[][] dp = new boolean[n + 1][m + 1];
//         dp[0][0] = true;
//         for (int i = 0; i < n; ++i) {
//             for (int j = 0; j <= m; ++j) {
//                 if (j < stones[i]) {
//                     dp[i + 1][j] = dp[i][j];
//                 } else {
//                     dp[i + 1][j] = dp[i][j] || dp[i][j - stones[i]];
//                 }
//             }
//         }
//         for (int j = m;; --j) {
//             if (dp[n][j]) {
//                 return sum - 2 * j;
//             }
//         }
//     }
// }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值