算法刷题Day 43 最后一块石头的重量II+目标和+一和零

Day 43 动态规划

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

注意第二个for循环那里不要漏了等于号

cpp 复制代码
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. 目标和

关键是要想到:

假设加法的总和为x,那么减法对应的总和就是sum - x。

所以我们要求的是 x - (sum - x) = target

x = (target + sum) / 2

此时问题就转化为,装满容量为x的背包,有几种方法

cpp 复制代码
class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        int sum = accumulate(nums.begin(), nums.end(), 0);
        if (abs(target) > sum) return 0;
        if ((sum - target) & 1)
        {
            return 0;
        }
        int bagSize = (sum + target) / 2;

        vector<int> dp(bagSize + 1, 0);
        dp[0] = 1; // 为什么等于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];
    }
};

474. 一和零

多了一个维度,但还是0-1背包问题

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

        for (auto str : strs)
        {
            int zeroNum = 0, oneNum = 0;
            for (auto ch : str)
            {
                if (ch == '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];
    }
};

总结

0-1背包的多种应用:

相关推荐
2301_7944615710 分钟前
详解七大排序
数据结构·算法·排序算法
爱coding的橙子1 小时前
蓝桥杯备赛 Day16 单调数据结构
数据结构·c++·算法·蓝桥杯
wuqingshun3141591 小时前
经典算法 约数之和
数据结构·c++·算法·蓝桥杯
溟洵1 小时前
【C/C++算法】蓝桥杯之递归算法(如何编写想出递归写法)
c语言·c++·算法
XYY3691 小时前
搜索与图论 树的深度优先遍历 树的重心
算法
破东风2 小时前
leetcode每日一题:替换子串得到平衡字符串
算法·leetcode·滑动窗口
Hole_up2 小时前
蓝桥杯真题-分糖果-题解
python·算法·职场和发展·蓝桥杯
泛舟起晶浪2 小时前
特殊的质数肋骨--dfs+isp
算法·深度优先
GGBondlctrl2 小时前
【leetcode】记录与查找:哈希表的题型分析
算法·力扣·两数之和·字母异位词分组·存在重复字符2
视觉AI2 小时前
研究下适合部署在jeston上的深度学习类单目标跟踪算法
深度学习·算法·目标跟踪