问题:
N个物品,容量为V的背包,每件物品只能使用一次。
第 i 件物品的体积是 vi ,价值是 wi 。
求解将哪些物品装入背包,可使这些物品的总体积不超过背包容量,且总价值最大。
输出最大价值。
状态表示:dp[i][j]表示只从前i个物品中选,且总体积不超过j的所有选法的最大价值。
状态计算:dp[i][j] = max(dp[i-1][j], dp[i-1][j-v[i]]+w[i])
说明:
dp[i][j]的所有选法可分成两类:选i,不选i
不选i:dp[i-1][j]
选i:dp[i-1][j-v[i]]+w[i] (退一步海阔天空)
朴素:
#include<bits/stdc++.h>
using namespace std;
const int N=1005;
int dp[N][N],w[N],v[N];
int main(){
int n,m;
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=0;j<=m;++j){
dp[i][j]=dp[i-1][j];
if(j>=v[i]) dp[i][j]=max(dp[i][j],dp[i-1][j-v[i]]+w[i]);
}
}
cout<<dp[n][m];
}
优化:
#include<bits/stdc++.h>
using namespace std;
const int N=1005;
int dp[N],w[N],v[N];
int main(){
int n,m;
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){
dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
}
}
cout<<dp[m];
}
如何优化的?
发现两个事实(优化的条件):
1. 计算dp[i][]只用到了dp[i-1][] ----> 滚动数组优化
2. 第二维(j 或 j-v[i])永远小于等于j ----> 可以优化成一维
第二维体积要从大到小枚举:
计算dp[i][j]需要用到的dp[i-1][j]和dp[i-1][j-v[i]]的第二维体积:j和j-v[i],都小于等于j,去掉第一维[i],要再搭配上第二维[j]倒着枚举,这样dp[j]才等于dp[i-1][j]、dp[j-v[i]才等于dp[i-1][j-v[i]],因为第i层,j从大到小枚举,小于等于j 的 j 和 j-v[i] 还没被计算过,就是第i-1层的值。
如果j从小到大枚举 用到的 dp[i-1][j-v[i]]
第二维j-v[i]是小于V的 也就是会用到前面已经计算过的值 那是新的第i层的值 不是i-1层的值
j从大到小枚举 用到的dp[j-v[i]]还没被更新过 所以就是要用的i-1层的值