代码随想录 第九章 动态规划part03 01背包问题 一维 416. 分割等和子集

01背包问题 一维

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
int main(){
    int n, bagWeight;
    cin >> n >> bagWeight;
    std::vector<int> value(n, 0);
    std::vector<int> weight(n, 0);
    for (int i = 0; i < n; i++) cin >> weight[i];
    for (int i = 0; i < n; i++) cin >> value[i];
    std::vector<int> result(bagWeight + 1, 0);
    for (int i = weight[0]; i<=bagWeight; i++) result[i] = value[0];
    for (int i = 1; i < n; i++){
        for (int j = bagWeight; j >= 0; j--){
            if (j >= weight[i]) result[j] = max(result[j], result[j - weight[i]] + value[i]);
        }
    }
    cout << result[bagWeight];
    return 0;
}

一维与二维的区别在于记录最优价值的数组是一维还是二维,从随想录所给的计算思路可以看出,在计算二维数组的过程中,只需要知道上一行的数据即可计算下一行的最优价值,所以完全可以只借助一个一维数组,不断迭代数组的值,求取最终的最优价值。

416. 分割等和子集

cpp 复制代码
class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum=0;
        for (int i = 0; i < nums.size(); i++) sum += nums[i];
        if (sum % 2 != 0) return false;
        vector<int> value((sum / 2) + 1, 0);
        for (int i = nums[0]; i < value.size(); i++) value[i] = nums[0];
        for (int i = 1; i < nums.size(); i++){
            for (int j = sum / 2; j>=0; j--){
                if (j >= nums[i]) value[j] = max(value[j], value[j - nums[i]] + nums[i]);
            }
        }
        if (value[sum / 2] != sum / 2) return false;
        return true;
    }
};

这题乍一看首先想到的思路就是回溯,不过现在是动态规划了。如果将数字集合看做重量与价值相等的物品集合,如果集合能分为两个和相等的子集,在背包问题中也就意味着在背包最大容量为所有物品重量和的1/2时,最多能装下重量为所有物品和的1/2的物品,因为物品价值与重量相等,所以最优价值不会大于背包容量,但如果最终最优价值低于背包容量,也就代表数字集合无法被分为两个和相等的部分。

代码随想录 第九章 动态规划part03

相关推荐
im_AMBER3 小时前
算法笔记 09
c语言·数据结构·c++·笔记·学习·算法·排序算法
凯芸呢3 小时前
Java中的数组(续)
java·开发语言·数据结构·算法·青少年编程·排序算法·idea
寂静山林3 小时前
UVa 1030 Image Is Everything
算法
AI柠檬3 小时前
几种排序算法的实现和性能比较
数据结构·算法·c#·排序算法
weixin_429630264 小时前
第6章 支持向量机
算法·机器学习·支持向量机
SweetCode4 小时前
C++ 实现大数加法
开发语言·c++·算法
王哈哈^_^4 小时前
【数据集】【YOLO】【目标检测】共享单车数据集,共享单车识别数据集 3596 张,YOLO自行车识别算法实战训推教程。
人工智能·算法·yolo·目标检测·计算机视觉·视觉检测·毕业设计
CodeWizard~5 小时前
AtCoder Beginner Contest 430赛后补题
c++·算法·图论
大大dxy大大5 小时前
机器学习-KNN算法示例
人工智能·算法·机器学习
关于不上作者榜就原神启动那件事6 小时前
模拟算法乒乓球
开发语言·c++·算法