01背包的方案数与二维费用问题

最新推荐文章于 2024-09-11 19:54:50 发布

Hanknet

最新推荐文章于 2024-09-11 19:54:50 发布

阅读量49

点赞数

分类专栏： leetCode 文章标签：动态规划 c++ 算法 leetcode Powered by 金山文档

本文链接：https://blog.csdn.net/Hanknet/article/details/129211965

版权

leetCode 专栏收录该内容

50 篇文章 0 订阅

订阅专栏

文章介绍了三个使用01背包动态规划算法解决的编程问题：最后一块石头的重量II、目标和以及一和零。每个问题都涉及求解最优解，通过构建动态规划表格，计算不同状态下的最大收益或可行方案数，最后给出时间复杂度和空间复杂度分析。

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

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

方法：01背包

class Solution {
    #define maxn 1510
    int dp[maxn];
public:
    int lastStoneWeightII(vector<int>& stones) {
        int sum = accumulate(stones.begin(), stones.end(), 0);
        int t = sum/2;
        int n = stones.size();
        for (int i = 0; i < stones.size(); ++i) {
            for (int j = t; j >= stones[i]; --j) {
                dp[j] = max(dp[j], dp[j-stones[i]] + stones[i]);
            }
        }
        return sum - dp[t] - dp[t];
    }
};

$时间复杂度O(n*m)，空间复杂度O(t)

494. 目标和

方法：01背包（求方案数）

class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        int sum = accumulate(nums.begin(), nums.end(), 0);
        if ((target+sum) % 2 == 1 || abs(target) > sum) return 0;
        int bagSize = (sum+target)/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(n*m)，空间复杂度O(target)

474. 一和零

方法：01背包（二维费用）

class Solution {
    #define maxn 110
    int dp[maxn][maxn];
public:
    int findMaxForm(vector<string>& strs, int m, int n) {

        for (auto& str : strs) {
            int one = 0, zero = 0;
            for (char& c : str) {
                if (c == '0') ++zero;
                else ++one;
            }
            for (int i = m; i >= zero; --i) {
                for (int j = n; j >= one; --j) {
                    dp[i][j] = max(dp[i][j], dp[i-zero][j-one] + 1);
                }
            }
        }
        return dp[m][n];
    }
};

$时间复杂度O(strs.size()*n*m)，空间复杂度O(n*m)