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

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

转换成capacity=sum/2的01背包

class Solution {
public:
    int lastStoneWeightII(vector& sts) {
        int sum=0;
        for(auto i:sts)sum+=i;
        int cap=sum/2;
        int n=sts.size();
        int dp[n][cap+1];
        
        memset(dp,0,sizeof dp);
        for(int i=0;i

空间优化
01背包的空间优化是j倒着来
for(int j=cap;j>=sts[i];j--)
完全背包的空间优化是正着来

class Solution {
public:
    int lastStoneWeightII(vector& sts) {
        int sum=0;
        for(auto i:sts)sum+=i;
        int cap=sum/2;
        int n=sts.size();
        int dp[cap+1];
        
        memset(dp,0,sizeof dp);
        for(int i=0;i=sts[i];j--){
                dp[j]=max(dp[j],dp[j-sts[i]]+sts[i]);
            }
        }

        return sum-2*dp[cap];
    }
};