代码随想录算法训练营第四十三天| 1049.最后一块石头的重量 II、494.目标和、474.一和零

1049 最后一块石头的重量 II

题目链接：1049
思路：问题可以转化为：将这堆石头分成重量尽可能相同的两组，这样剩下重量最小。因此，可以转化为背包容量为sum/2，价值等于重量的0-1背包问题。当求出背包容量为sum/2时可装的最大质量dp[target]后，另一堆的质量为sum-dp[target]，由于sum/2向下取整，因此最后剩下的石头重量为sum-dp[target]-dp[target]。
代码：

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int sum = accumulate(stones.begin(), stones.end(), 0);
        int target = sum / 2;

        vector<int> dp(target+1, 0);
        for (int i=0; i<stones.size(); ++i) {
            for (int j=target; j>=stones[i]; --j) {
                dp[j] = max(dp[j], dp[j-stones[i]]+stones[i]);
            }
        }
        return (sum-dp[target])-dp[target];
    }
};

494 目标和

题目链接：494
思路：题目等价于将数组分为两组A和B，使得A-B=target。由于A+B=sum，因此A=(sum+target)/2，问题转换为，装满容量为(sum+target)/2的背包有几种方法。

1. 定义dp[j]代表填满容量为j的背包有dp[j]种方法。
2. 要填满容量为j的背包有两种方案：若不装nums[i]，则有dp[i-1][j]种方法，即上一层得到的dp[j]；若装i，则等价于用[0,i-1]填满容量为j-nums[i]的背包，再装一个nums[i]，即上一层的dp[j-nums[i]]，得递推公式dp[j]+=dp[j-nums[i]]。
3. 初始化时，由于是从i=0，即只考虑nums[0]开始遍历，因此对应的上一层就是一个数都不取，此时容量为0有1种方法（什么都不装），其他容量都是0种方法，因此dp[0]=1。
4. 遍历顺序同滚动数组的0-1背包。
   代码：

class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        int sum = accumulate(nums.begin(), nums.end(), 0);
        if ((target+sum)%2 != 0) return 0;
        if (abs(target) > sum) return 0;

        sum = (target + sum) / 2;
        vector<int> dp(sum+1, 0);
        dp[0] = 1;
        for(int i=0; i<nums.size(); ++i) {
            for (int j=sum; j>=nums[i]; --j) {
                dp[j] += dp[j-nums[i]];
            }
        }
        return dp[sum];
    }
};

474. 一和零

题目链接：474
思路：该问题也是0-1背包问题，只不过重量有两个维度m和n，每个物品的价值为1，即总价值为物品数量。因此，定义dp[i][j]代表当最多有i个0和j个1时，最大子集的长度（对应于0-1背包问题中的一维滚动数组）。递推公式dp[i][j]=max(dp[i][j], dp[i-zero_num][j-one_num]+1。
代码：

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        vector<vector<int>> dp(m+1, vector<int>(n+1, 0));
        for (int s=0; s<strs.size(); s++) {
            int zero_num = count(strs[s].begin(), strs[s].end(), '0');
            int one_num = count(strs[s].begin(), strs[s].end(), '1');
            for (int i=m; i>=zero_num; i--) {
                for (int j=n; j>=one_num; j--) {
                    dp[i][j] = max(dp[i][j], dp[i-zero_num][j-one_num]+1);
                }
            }
        }
        return dp[m][n];
    }
};