首先01背包问题问题的状态计算公式为:f[i][j]=max(f[i-1][j],f[i-1][j-v[i]]+w[i])
其中,f[i][j]表示从前i个物品中选择,最大容量不超过j,物品体积为v[i],价值为w[i]
在这里,我们需要明确一个问题,01背包的优化并不是优化时间复杂度,而是优化空间复杂度,是将f[i][j]从二维数组压缩为一维数组f[j]
1.为什么是将i这一维删除呢?
f[i][j]=max(f[i-1][j],f[i-1][j-v[i]]+w[i]),从公式中我们可以看出,在求f[i][j]的时候我们只需要用到上一层i的值,在双层循环中,因为还没更新,所以当前在第i层的时候,此时保留的就是上一层i-1的值,所以可以将i-1这一维舍去。
2.重点来了:为什么for j循环中,j需要从大到小开始循环呢?
未优化前的双层for循环:
for(int i=1;i<=n;i++){
for(int j=0;j<=m;j++){
f[i][j]=f[i-1][j];
if(j>=v[i]) f[i][j]=max(f[i-1][j],f[i-1][j-v[i]]+w[i]);
}
}
倘若优化后的双层for循环是这样的:
for(int i=1;i<=n;i++){
for(int j=0;j<=m;j++){
f[j]=f[j];
if(j>=v[i]) f[j]=max(f[j],f[j-v[i]]+w[i]);
}
}
f[i][j]=f[i-1][j] 可以看作是f[j]=f[j],因为等号后面的那个f[j]的值就是f[i-1][j],是还没更新的i-1层的j,而前面的f[j]就是这一层的f[i][j],是等待被更新的,所以可以直接去掉i这一维;
我们可以观察一下此时f[j-v[i]]是f[i-1][j-v[i]]还是f[i][j-v[i]]
因为j-v[i] 是小于 j 的,所以在第i层的时候,在j从小到大循环中,会先遍历到j-v[i],那么此时f[j-v[i]]就是f[i][j-v[i]]而不是f[i-1][j-v[i]],显然这样的遍历是错误的!
正确循环:
for(int i=1;i<=n;i++){
for(int j=m;j>=0;j--){
f[j]=f[j];
if(j>=v[i]) f[j]=max(f[j],f[j-v[i]]+w[i]);
}
}
此时,j会比j-v[i]先遍历到,f[j-v[i]]只能是上一层f[i-1][j-v[i]],这样才符合推出来的状态计算公式!
完整C++代码:
#include<iostream>
#include<cstdio>
using namespace std;
const int N=1005;
int n,m; //n表示物品总数,m表示背包总容量
int f[N],v[N],w[N];
//f[x]表示背包容量为x时的最大价值,v[]表示物品单个体积,w[]表示物品单个价值
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>v[i]>>w[i];
for(int i=1;i<=n;i++){
for(int j=m;j>=v[i];j--){
f[j]=max(f[j],f[j-v[i]]+w[i]);
}
}
cout<<f[m];
return 0;
}