【算法与数据结构】1049、LeetCode 最后一块石头的重量 II

最新推荐文章于 2024-08-25 22:13:13 发布

晚安66

最新推荐文章于 2024-08-25 22:13:13 发布

阅读量1k

点赞数 23

分类专栏：算法文章标签：算法

本文链接：https://blog.csdn.net/qq_45765437/article/details/135711305

版权

算法专栏收录该内容

129 篇文章 2 订阅

订阅专栏

本文解析了LeetCode中的一个问题，通过01背包方法解决最小石头粉碎值II，即在保证每堆石头质量尽可能接近的情况下，求两堆石头粉碎后的最小质量差。文章详细介绍了解题思路、程序实现以及复杂度分析。

摘要由CSDN通过智能技术生成

所有的LeetCode题解索引，可以看这篇文章——【算法和数据结构】LeetCode题解。

一、题目

在这里插入图片描述

二、解法

思路分析：本题需要得到石头之间两两粉碎之后的最小值，那么一个简单的思路就是将这堆石头划分成大小相近的两小堆石头，然后粉碎，这样得到的结果必然是最优值。那么如何划分呢？我们可以将所有石头的质量求和，假设和为 $s u m$ ，以 $s u m /2$ 为界限。一堆石头质量设为 $w_1$ ，有 $w_1 \leq sum/2$ ；而另外一堆石头质量 $w_2$ ，有 $w_2 \geq sum/2$ ，且 $w_1 + w_2=sum$ 。因为要求两堆石头粉碎之后的质量 $\Delta$ 最小，所以划分出来的两堆石头重量越接近越好，等同于 $w_1$ 和 $w_2$ 越接近于 $s u m /2$ 。所以我们可以将 $min(\Delta =w_2- w_1)$ 问题，转化为 $max(w_1), s.t. w_1 \leq sum/2$ ，即在这个数组中找到和最接近sum/2的子集。这是一个01背包问题。最终的最小质量 $\Delta_{min}= w_2- w_1 = sum - 2*w_1$ 。
程序如下：

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int sum = accumulate(stones.begin(), stones.end(), 0);
        vector<int> dp(vector<int>(sum/2 + 1, 0));
        for (int i = 0; i < stones.size(); i++) {			// 遍历物品
            for (int j = sum/2; j >= stones[i]; j--) {			// 遍历背包容量
                dp[j] = max(dp[j], dp[j - stones[i]] + stones[i]);
            }
        }
        return sum -2 * dp[sum/2];
    }
};

复杂度分析：

时间复杂度： $O(n^2)$ 。
空间复杂度： $O (n)$ 。

三、完整代码

# include <iostream>
# include <vector>
# include <numeric>
# include <algorithm>
using namespace std;

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int sum = accumulate(stones.begin(), stones.end(), 0);
        vector<int> dp(vector<int>(sum/2 + 1, 0));
        for (int i = 0; i < stones.size(); i++) {			// 遍历物品
            for (int j = sum/2; j >= stones[i]; j--) {			// 遍历背包容量
                dp[j] = max(dp[j], dp[j - stones[i]] + stones[i]);
            }
        }
        return sum -2 * dp[sum/2];
    }
};

int main() {
    Solution s1;
    vector<int> stones = { 2,7,4,1,8,1 };
    int result = s1.lastStoneWeightII(stones);
    cout << result << endl;
    system("pause");
    return 0;
}