代码随想录算法训练营第四十三天|LeetCode1049. 最后一块石头的重量 II

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

题目链接：1049. 最后一块石头的重量 II
用背包问题的解题方法去解决。
首先求出石头的总重量sum，然后用它长度和sum/2去创建一个二维数组，即dp[stones.size()][sum/2];
初始化数组:

for(int i = stones[0]; i <= sum/2; i++) {
            dp[i] = stones[0];
        }

递推公式：
dp[i][j]由两个方向推到过来，即dp[i -1][j]和dp[i - 1][j - stones[i]] + stones[i]。
所以dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - stones[i]] + stones[i])
确定遍历顺序:

for(int i = 1; i < stones.size(); i++) {
            for(int j = sum; j >= stones[i]; 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]);
            }
        }

最后得到的dp数组的最后一个数就是最接近sum/2的和石头重量之和。
于是最后剩余的石头的重量应该是sum - dp[stones.size() - 1][sum/2]。
代码如下：

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int sum = 0; 
        //先求出石头的总重量
        for(int i = 0; i < stones.size(); i++) {
            sum += stones[i];
        }

        vector<vector<int>>dp(stones.size(), vector<int>(sum/2 + 1, 0));//创建数组

        for(int i = stones[0]; i <= sum/2; i++) {//初始化
            dp[0][i] = stones[0];
        }

        for(int i = 1; i < stones.size(); i++) {//确定遍历顺序
            for(int j = 0; j <= sum/2; 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 sum - 2 * dp[stones.size() - 1][sum/2];

    }
};