代码随想录算法训练营第三十天 | 01背包问题 二维 01背包问题 一维 416. 分割等和子集

  1. 携带研究材料(第六期模拟笔试)
题目描述

小明是一位科学家,他需要参加一场重要的国际科学大会,以展示自己的最新研究成果。他需要带一些研究材料,但是他的行李箱空间有限。这些研究材料包括实验设备、文献资料和实验样本等等,它们各自占据不同的空间,并且具有不同的价值。

小明的行李空间为 N,问小明应该如何抉择,才能携带最大价值的研究材料,每种研究材料只能选择一次,并且只有选与不选两种选择,不能进行切割。

输入描述

第一行包含两个正整数,第一个整数 M 代表研究材料的种类,第二个正整数 N,代表小明的行李空间。

第二行包含 M 个正整数,代表每种研究材料的所占空间。

第三行包含 M 个正整数,代表每种研究材料的价值。

输出描述

输出一个整数,代表小明能够携带的研究材料的最大价值。

输入示例
复制代码
6 1
2 2 3 1 5 2
2 3 1 5 4 3
输出示例
复制代码
5
cpp 复制代码
#include<bits/stdc++.h>
using namespace std;

int n,bagweight;

void solvation()
{
    std::vector<int> weight(n,0);
    vector<int> value(n,0);
    for(int i = 0 ; i < n; i++)
    {
        cin >> weight[i];
    }
    for(int i = 0 ; i < n ; i++)
    {
        cin >> value[i];
    }
    
    vector<vector<int>> dp(weight.size(),vector<int>(bagweight+1,0));
    
    for(int i = weight[0] ; i <= bagweight ; i ++)
    {
        dp[0][i] = value[0];
    }
    
    for(int i = 1; i < weight.size() ; i++)
    {
        for(int j = 0; j <= bagweight ; j ++)
        {
            if(weight[i] > j) dp[i][j] = dp[i-1][j];
            else dp[i][j] = max(dp[i-1][j],dp[i-1][j - weight[i]]+ value[i] );
        }
    }
    
    cout << dp[weight.size() - 1][bagweight];
}


int main()
{
    while(cin >> n >> bagweight)
    {
        solvation();
    }
    
    return 0 ;
}

滚动数组的写法:

cpp 复制代码
#include<bits/stdc++.h>
#include<vector>
using namespace std;

int n ,bagweight;

void solveproblem()
{
    std::vector<int> weight(n);
    vector<int> value(n);
    
    for(int i = 0; i < n; i++)
    {
        cin >> weight[i];
    }
    for(int i = 0; i < n; i++)
    {
        cin >> value[i];
    }
    
    vector<int> dp(bagweight+1,0);
    
    for(int i = 0 ; i < weight.size(); i++)
    {
        for( int j = bagweight ; j >= weight[i] ; j--)
        {
            if(j < weight[i]) dp[j] = dp[j-1];
            else dp[j] = max(dp[j],dp[j - weight[i]] + value[i]);
        }//注意这里不是max(dp[j-1])
    }
    
    cout << dp[bagweight] <<endl;
}

int main()
{
    while(cin >> n >> bagweight)
    {
        solveproblem();
    }
    return 0 ;
}

416. 分割等和子集

给你一个 只包含正整数非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。

示例 1:

复制代码
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。

示例 2:

复制代码
输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集
cpp 复制代码
class Solution {
public:
    bool canPartition(vector<int>& nums) 
    {
        int sum = 0;
        vector<int> dp(20000,0);
        for(int i = 0 ; i < nums.size() ; i++)
        {
            sum += nums[i];
        }
        if(sum%2 == 1) return false;

        int mid = sum/2;

        for(int i = 0 ; i < nums.size(); i++)
        {
            for(int j = mid ; j >= nums[i] ; j--)
            {
                dp[j] = max(dp[j],dp[j - nums[i]] + nums[i]);
            }
        }
        if(dp[mid] == mid) return true;
        return false;
    }
};
相关推荐
许小燚3 小时前
线性表——双向链表
数据结构·链表
董董灿是个攻城狮3 小时前
5分钟搞懂什么是窗口注意力?
算法
Dann Hiroaki3 小时前
笔记分享: 哈尔滨工业大学CS31002编译原理——02. 语法分析
笔记·算法
xiaolang_8616_wjl4 小时前
c++文字游戏_闯关打怪2.0(开源)
开发语言·c++·开源
夜月yeyue4 小时前
设计模式分析
linux·c++·stm32·单片机·嵌入式硬件
qqxhb5 小时前
零基础数据结构与算法——第四章:基础算法-排序(上)
java·数据结构·算法·冒泡·插入·选择
无小道5 小时前
c++-引用(包括完美转发,移动构造,万能引用)
c语言·开发语言·汇编·c++
晚云与城5 小时前
【数据结构】顺序表和链表
数据结构·链表
FirstFrost --sy7 小时前
数据结构之二叉树
c语言·数据结构·c++·算法·链表·深度优先·广度优先
森焱森7 小时前
垂起固定翼无人机介绍
c语言·单片机·算法·架构·无人机