每日一题 LeetCode1049. 最后一块石头的重量 II java题解

题目

https://leetcode-cn.com/problems/last-stone-weight-ii/
背包问题可参考：https://leetcode-cn.com/problems/last-stone-weight-ii/solution/yi-pian-wen-zhang-chi-tou-bei-bao-wen-ti-5lfv/

思路

石头总重量为sum,取负的重量为neg，另外的为sum-neg。最后一块石头的重量为sum-2neg。
要使最后一块石头的重量最小，neg要最大（但不大于sum/2）。

将石头分为两堆，使他们都尽可能接近sum/2(sum是石头总和)。

使用01背包的方法.
dp数组：
dp[i+1][target]表示前i个（0到i-1）石头能否达到target重量
初始状态：
dp[0][0]为真，dp[0][其他]为假。
转移方程：
stone[i]<=target,dp[i+1][target]=dp[i][target]或dp[i][target-stone[i]]
stone[i]>target,dp[i+1][target]=dp[i][target]

求出dp[n][]后，，所有为真的 dp[n][j] 中带入sum-neg就可得到最后一块石头的最小值。

代码

class Solution {
    public int lastStoneWeightII(int[] stones) {
        int len=stones.length;
        int sum=0;
        for(int i=0;i<len;i++){
            sum+=stones[i];
        }
        boolean[][] dp=new boolean[len+1][sum/2+1];
        dp[0][0]=true;
        for(int i=0;i<len;i++){
            for(int target=0;target<=sum/2;target++){
                if(stones[i]<=target)
                    dp[i+1][target]=(dp[i][target]||dp[i][target-stones[i]]);
                else
                    dp[i+1][target]=dp[i][target];
            }
        }
        for(int target=sum/2;target>=0;target--){
            if(dp[len][target])
                return sum-2*target;
        }
        return 0;
    }
}

复杂度

时间复杂度：O(n⋅sum)。其中 n 是数组 stones 的长度，sum 为 stones 所有元素之和。
空间复杂度：O(n*sum)。

可以空间优化到O(sum)

但是内层循环要倒序遍历。