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

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

题目

解法

题解链接
1.

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        vector<int> dp(15001, 0);
        int sum = 0;
        for (int i = 0; i < stones.size(); i++) sum += stones[i];
        int target = sum / 2;
        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];
    }
};

时间复杂度：O( m*n)
空间复杂度：O( m)

494. 目标和

题目

解法

题解链接
1.

class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        int sum = 0;
        for (int i = 0; i < nums.size(); i++) sum += nums[i];
        if (abs(target) > sum) return 0; // no plan
        if ((target+sum) % 2 == 1) return 0; // no plan
        int bagSize = (target + sum) / 2;
        vector<int> dp(bagSize+1, 0);
        dp[0] = 1;
        for (int i = 0; i < nums.size(); i++) {
            for (int j = bagSize; j >= nums[i]; j--) {
                dp[j] += dp[j-nums[i]]; 
            }
        }
        return dp[bagSize];
    }
};

时间复杂度：O(m*n )
空间复杂度：O( m) m:背包容积

474. 一和零

题目

解法

题解链接
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(string str : strs){
            int zeroNum = 0;
            int oneNum = 0;
            for(char a : str){
                if(a =='0') zeroNum++;
                else oneNum++;
            }
            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];
    }
};

时间复杂度：O(kmn )
空间复杂度：O(mn )

感悟

动态规划还需要深入了解