训练营打卡Day43

训练营打卡Day43

题100：1049. 最后一块石头的重量 II

思路

• 首先算出石头总重量，然后将目标值设为总重量的一半。
• 接着使用01背包问题的思路和方法，通过循环遍历石头数组中的每个石头，每次更新dp数组的值，dp[j] = max(dp[j], dp[j - stones[i]] + stones[i])，其中dp[j]表示当前重量为j时，所能获得的最大重量。
• 最后返回总重量减去dp[target]*2

代码

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int sum = accumulate(stones.begin(), stones.end(), 0);
        int target = sum/2;
        vector<int>dp(10001, 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];
    }
};

101：494. 目标和

思路

• 计算nums数组中所有元素的总和，然后判断如果总和小于目标值的绝对值或者总和加上目标值不能被 2 整除，那么显然无解，直接返回0
• 接着将目标值转化为 s = (sum + target) / 2
• 然后使用01背包问题的思路和方法，通过循环遍历nums数组中的每个元素，每次更新dp数组的值，dp[j] += dp[j - nums[i]]，其中dp[j]表示当前重量为j时，所能获得的方案数。
• 最后返回dp[s]即为答案，表示有多少种方案可以和为target.

代码

class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {

        int sum = accumulate(nums.begin(), nums.end(), 0);
        if(sum < abs(target) || (sum + target) % 2 == 1) return 0;
        int s = (sum + target) / 2;
        vector<int>dp(s+1, 0);
        
        dp[0] = 1;
        for(int i = 0; i < nums.size(); i++)
            for(int j = s; j >= nums[i] ; j--)
                dp[j] += dp[j - nums[i]];  
        return dp[s];
    }
};

题102：474. 一和零

思路

• 首先声明一个二维dp数组，用来记录每种状态下的最大值。
• 然后遍历字符串数组，对于每个字符串，统计出其中0和1的数量。
• 接着使用01背包问题的思路和方法，通过循环遍历dp数组中的每个元素，每次更新dp数组的值，dp[i][j] = max(dp[i][j], dp[i-zeroNum][j-oneNum]+1)。其中dp[i][j]表示当前最大数量为i和j时，所能获得的最大数量。
• 最后返回dp[m][n]即为答案，表示最多能选取多少个字符串。

代码

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        vector<vector<int>>dp(m+1, vector<int>(n+1, 0));

        for(const string& str: strs)
        {
            int zeroNum = 0, oneNum = 0;
            
            for(const char& ch: str)
            {
                if(ch == '1') oneNum++;
                else zeroNum++;
            }

            for(int i = m; i >= zeroNum; i--)
                for(int j = n; j >= oneNum; j--)
                    dp[i][j] = max(dp[i][j], dp[i-zeroNum][j-oneNum]+1);
        }
        return dp[m][n];
    }
};