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

题目

分析

本题和分割等和子集题目有异曲同工之妙。本题就是将石头尽可能分成重量相同的两堆，使得这两堆重量相减之后结果最小。这就转化为了分割相同等和子集的题目。

代码

二维dp数组

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int sum = 0;
        for(auto s:stones) sum += s;
        int tmp = sum;
        sum /= 2;
        int dp[stones.size()][sum+1];
        //初始化
        memset(dp,0,sizeof(dp));
        for(int j = 0;j <= sum;j++){
            if(j >= stones[0]) dp[0][j] = stones[0];
        }
        for(int i = 1;i < stones.size();i++){
            for(int j = 0;j <= sum;j++){
                if(j < stones[i] ) dp[i][j] = dp[i-1][j];
                else dp[i][j] = max(dp[i-1][j],dp[i-1][j-stones[i]] + stones[i]);
            }
        }
        return tmp - 2 * dp[stones.size()-1][sum] ;
    }
};

一维dp数组

动态转移公式  dp[j] = max(dp[j],dp[j-stones[i]] + stones[i]);

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        vector<int>dp(30*1000/2 + 1,0); //30*1000/2是最大容量
        int sum = 0;
        for(auto s:stones){
            sum += s;
        }
        int tmp = sum;
        sum /= 2;
        for(int i = 0;i < stones.size();i++){
            for(int j = sum;j >= stones[i];j--){
                dp[j] = max(dp[j],dp[j-stones[i]] + stones[i]);
            }
        }
        return tmp - 2*dp[sum];
    }
};