算法学习——01背包问题

       最近在考研，学习数据结构时突然意识到自己的算法基础有点太薄弱，遂决定定期学习一些算法题，并记录在这里，微小见解，如有错误，欢迎各位大佬评论区批评指正。

题目如下：

状态转移方程及其解释

f[i][j]表示在背包容量在j的情况下，前i个物品的价值

(1)式表示第i个物品未装入背包，此时背包的价值仍为第i-1个物品时的最大价值；

(2)式表示第i个物品装入背包，此时背包的价值为第i-1个物品时的价值+第i个物品的价值(ps:这种情况下，背包的剩余容量必须小于等于第i个物品的价值)。

代码实现

1.未优化版本的实现

        该版本下使用二维数组

//未优化算法——采用二维数组
#include<iostream>
#include<cmath>
using namespace std;

const int N=1010;

int n,m;//物品数量，背包容积
int v[N],w[N];//第i个物品的体积和价值
int f[N][N];//背包中物品的价值

int main(){
    cin>>n>>m; //输入第一行的两个整数
    for(int i=1;i<=n;i++){ //输入第i个物品的体积和价值
        cin>>v[i]>>w[i];
    }
    
    for(int i = 1;i<=n;i++){
        for(int j=0;j<=m;j++){
            //不取第i个物品
            f[i][j]=f[i-1][j];
            //如果背包容量大于等于第i个物品的体积，就要考虑取第i个物品的情况
            if(j>=v[i]){
                f[i][j]=max(f[i][j],f[i-1][j-v[i]]+w[i]);
            }
        }
    }
    cout<<f[n][m]<<endl;
    return 0;
}

2.优化版本 

        采用一维数组

        为什么可以使用一维数组呢？我们定义的状态f[i][j]可以求得任意合法的i与j最优解，但题目只需要求得最终状态f[n][m]，因此我们只需要一维的空间来更新状态。

//优化算法——采用一维数组
#include<iostream>
#include<cmath>
using namespace std;

const int N=1010;

int n,m;//物品数量，背包容积
int v[N],w[N];//第i个物品的体积和价值
int f[N];//容积为N时背包中物品的价值

int main(){
    cin>>n>>m; //输入第一行的两个整数
    for(int i=1;i<=n;i++){ //输入第i个物品的体积和价值
        cin>>v[i]>>w[i];
    }
    
    for(int i = 1;i<=n;i++){
        for(int j=m;j>=v[i];j--){
            //未选第i个物品时，价值仍是f[j]
            //选择第i个物品时，价值为背包内原有价值+第i个物品的价值
            f[j]=max(f[j],f[j-v[i]]+w[i]); 
            
        }
    }
    cout<<f[m]<<endl;
    return 0;
}

代码解释：

为什么二重循环中j需要逆序：

因为：此时f[j-v[i]] ≠ 二维数组f[i-1][j-v[i]]+w[i]:
理由：通过第二个for顺序循环j按顺序从0到m的话，f[j] = 二维数组的f[i][j - v[i]]+w[i]，【比如dp[3] = max(dp[3], dp[2] + 1);因为正序j从0->m ,dp[j]是从小到大更新，j=3中dp[2]已经更新成的前i个物品dp[2],但此时我们要求的是前i-1个物品下的dp[2]】，所以只有逆序第二个for的j进行循环时f[j-v[i]]，dp[j]从大到小更新，较小的数还是二维数组的f[i-1][j-v[i]]+w[i]。【比如dp[8] = max(dp[8], dp[6] + 2)当更新dp[8]时，在这第二个for循环中dp[6]还没有被更新，还是第一个for循环中前i-1个物品的dp[6]】.这样才不是被污染的数据。