049. 最后一块石头的重量 II 力扣每日一题 C++题解

欢迎关注微信公众号：【没伞的孩子拼命跑】 和我的小伙伴们一块打卡吧

题目要求： 1049. 最后一块石头的重量 II

题目解析

题目很简单，就是给你一堆石头，让你两个两个的废碎，如果是一样大的石头就完全粉碎，否则就求两个的差接着粉碎。问到最后剩下的时候大小最小是多少？

解题思路

方法一：

这个题目不难，因为这个月是背包月，大家可能能想到背包，但是如果是面试的时候，你能想到吗？一般来讲，这个题我肯定是先想到的贪心，不行之后才想到的背包问题。

首先写一个错误的思路，就是贪心，我每次都选择最大的两个进行粉碎，是不是最后就最小了呢？当然这是不行的，局部最优无法得到全局最优。例如 [31,26,33,21,40] 如果按照是贪心结果应该是9，而真实的结果应该是5.

代码和详细注释如下

//C++题解
class Solution {
public://贪心的解法
    int lastStoneWeightII(vector<int>& stones) {
        priority_queue<int>q;//建立一个优先队列
        for(int i = 0;i < stones.size();i++)
        {
            q.push(stones[i]);
        }
        while(q.size()!=1)//每次选择最大的两个进行粉碎
        {
            int a = q.top();
            q.pop();
            int b = q.top();
            q.pop();
            if(a!=b)
            {
                q.push(abs(a-b));
            }
            if(q.size()==0)break;
        }
        return q.empty()?0:q.top();
    }
};

方法二：

如果说我们仔细想象这个问题，为了使得最后的结果最小，我们是不是可以分成两堆，然后这两堆石头重量之差的绝对值也是所有划分当中最小的时候，是不是粉碎之后的结果就最小。那么问题就转为如何分石头使得两个的差最小，这不就可以转为01背包问题了吗，每个石头你可以选，也可以不选，看背包为一半的时候你最多能装多少，装的越多，剩下的越少，两者的差也就越小。

//C++题解
class Solution {
public://动态规划dp[i][j]表示i个物品装到j里面，最多能装多少。
    int lastStoneWeightII(vector<int>& stones) {
        int len = stones.size();
        int sum = 0;
        for(int i = 0;i < len; i++)
        {
            sum+=stones[i];
        }
        int mid = sum/2;
        vector<vector<int>>dp(len+1,vector<int>(mid+1,0));
        for(int i = 1;i <= len; i++)
        {
            for(int j = 0 ;j <= mid; j++)
            {
                dp[i][j] = dp[i-1][j];
                if(j >= stones[i-1])
                    dp[i][j] = max(dp[i-1][j],dp[i-1][j-stones[i-1]]+stones[i-1]);
            }
        }
        return sum-2*dp[len][mid];
    }
};

当然这个还是可以优化为一维的，具体方法可以看我昨天发的公众号。 通过情况