代码随想录算法训练营第四十一天 | 01背包问题 二维

本文解释了01背包问题的基本概念,介绍了如何通过C++代码实现二维动态规划来求解,包括变量定义、数组初始化和优化的循环结构策略。
摘要由CSDN通过智能技术生成

01背包问题,说实话还是很陌生,而且这两天感冒,头晕晕的,什么都没看进去。

#include<bits/stdc++.h>
using namespace std;

int main(){
    int size,space;
    cin>>size>>space;
    vector<int> weight(size);
    vector<int> val(size);
    
    for(int i=0;i<size;i++){
        cin>>weight[i];
    }
    
    for(int i=0;i<size;i++){
        cin>>val[i];
    }
    
    vector<vector<int>> dp(weight.size(),vector<int>(space+1,0));
    
    for(int j=weight[0];j<=space;j++){
        dp[0][j]=val[0];
    }
    
    for(int i=1;i<size;i++){//遍历科研物品
        for(int j=0;j<=space;j++){//遍历背包容量
            if(j<weight[i]) dp[i][j]=dp[i-1][j];
            else dp[i][j]=max(dp[i-1][j],dp[i-1][j-weight[i]]+val[i]);
        }
    }
    cout<<dp[size-1][space]<<endl;
    return 0;
}

先明确概念,size是指物品的个数,space是指背包的空间。

weight是指某一个物品的重量,val是指某一个物品的价值。

我们建立一个二维dp数组,其中二维数组的 i 表示物品 i ,所以 i 的size 就是等于物品个数。

二维数组的 j 就是指背包的容量大小, 也就是背包的空间。这个可能有点难理解,我们把背包的空间大小设为 j 的size,也就是 当 j 取到最大的时候,这时候 j==space;  这样子做,我们在解题的时候,j 是随着循环变大直到变得最大, 方便我们做一个分割。

然后我们队dp数组进行初始化,先全部初始化为0,然后对第一行,也就是i==0进行初始化,如果背包空间大小足以装得下第一件物品,那么我们就把第一行中背包大小足够的那几个位置都初始化为第一件物品的价值。这样是因为我们dp数组的第一行,代表就是只能装一件物品,而且是只能是第一件物品。所以我们就这样子初始化,方便后面进行操作。

在对dp数组赋值的二重循环中,先遍历科研物品,因为上面已经对第一件物品进行初始化,所以我们从第二件开始。然后在内层循环不断将背包空间大小扩大,直到上限,看看能否放下第二件物品,如果能的话那么我们就把当前背包大小减去第二件物品大小,得到的就是剩余空间,如果剩余空间能放得下第一件物品,那么我们就可以直接将第一件物品和第二件物品相加的价值放进dp数组;如果剩余空间放不下第一件物品,那么我们就比较价值。这里在循环里面的意思差不多是这样子,但是写出来的代码用了比较巧妙的方法。通过不断扩大 j ,然后 j 减去weight的剩余也肯定是不断增大的,我们取数组的对应位置,这个位置肯定是我们在前面就计算出的最大价值的位置,这样子就取得一个最大价值,然后在加上这个val,得出的就有可能是最优结果。

  • 8
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值