2021.9.7 力扣-最后一块石头的重量 II

同样是01背包的应用，但还是完全想不出怎么才能将其转化为01背包问题，在我看来这不完全是两样东西？于是到题解区看了看提示，看到了这句话：

 瞬间豁然开朗！

题目描述：

有一堆石头，用整数数组 stones 表示。其中 stones[i] 表示第 i 块石头的重量。

每一回合，从中选出任意两块石头，然后将它们一起粉碎。假设石头的重量分别为 x 和 y，且 x <= y。那么粉碎的可能结果如下：

如果 x == y，那么两块石头都会被完全粉碎；
如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。
最后，最多只会剩下一块 石头。返回此石头 最小的可能重量 。如果没有石头剩下，就返回 0。

方法一：

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int n = stones.size();
        int sum = 0;
        for (int i = 0; i < n; i++)
        {
            sum += stones[i];
        }
        vector<vector<int>> f(n , vector<int>(sum / 2 + 1, 0));
        //f[i][j]表示在下标为0 - i的元素中，选择某些元素，使其和（类比物品总重量）不超过j，且和（类比物品的总价值）最大
        for (int j = sum / 2; j >= stones[0]; j--)
        {
            f[0][j] = stones[0];
        }
        for (int i = 1; i < n; i++)
        {
            for (int j = 1; j <= sum / 2; j++)
            {
                if (j < stones[i])
                {
                    f[i][j] = f[i - 1][j];
                }
                else
                {
                    f[i][j] = max(f[i - 1][j], f[i - 1][j - stones[i]] + stones[i]);
                }
            }
        }
        return sum - 2 * f[n - 1][sum / 2];
    }
};

设f[i][j]表示在下标为0 - i的元素中，选择某些元素，使其和（类比物品总重量）不超过j，且和（类比物品的总价值）最大，那么接下来就可以完全利用01背包问题的模板来写了。

最后返回的时候，返回的实际上是（sum - f[n - 1][sum / 2]）- f[n - 1][sum / 2]，也就是将数组stones中的另外一半减去所得到的一半，由于当sum为偶数时两部分的元素之和相等，当sum为奇数时sum/2会舍去0.5，所以根据f[i][j]的定义，（sum - f[n - 1][sum / 2]）这一部分肯定大于等于f[n - 1][sum / 2]。

方法一的空间优化：

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int n = stones.size();
        int sum = 0;
        for (int i = 0; i < n; i++)
        {
            sum += stones[i];
        }
        vector<int> f(sum / 2 + 1, 0);
        //f[j]表示选择某些元素，使其和（类比物品总重量）不超过j，且和（类比物品的总价值）最大
        for (int j = sum / 2; j >= stones[0]; j--)
        {
            f[j] = stones[0];
        }
        for (int i = 1; i < n; i++)
        {
            for (int j = sum / 2; j >= stones[i]; j--)
            {
                f[j] = max(f[j], f[j - stones[i]] + stones[i]);
            }
        }
        return sum - 2 * f[sum / 2];
    }
};

老方法，将二维vector改为一维vector。

• 时间复杂度：O(m * n) , m是石头总重量（准确的说是总重量的一半），n为石头块数
• 空间复杂度：O(m)