背包常见题目一
有
N
N
N件物品和一个容量是
V
V
V 的背包。每件物品只能使用一次。
第
i
i
i件物品的体积是
v
i
v_i
vi,价值是
w
i
w_i
wi。
求解将哪些物品装入背包,可使这些物品的总体积不超过背包容量,且总价值最大。
输出最大价值。
数据范围
0
<
N
,
V
≤
1000
0<N,V≤1000
0<N,V≤1000
0
<
v
i
,
w
i
≤
1000
0<v_i,w_i≤1000
0<vi,wi≤1000
朴素写法
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1010;
int n,m;
int dp[maxn][maxn];
int v[maxn],w[maxn];
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=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];
return 0;
}
时间复杂度: O ( n m ) O(nm) O(nm) 空间复杂度: O ( n 2 ) O(n^2) O(n2)
空间初步优化
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1010;
int n,m,v,w;
int dp[maxn][maxn];
int main(){
cin >> n >> m;
for(int i=1;i<=n;i++){
cin >> v >> w;
for(int j=0;j<=m;j++){
dp[i][j] = dp[i-1][j]; // 不取当前物品
if(j>=v) dp[i][j] = max(dp[i][j],dp[i-1][j-v]+w);
}
}
cout << dp[n][m];
return 0;
}
时间复杂度: O ( n m ) O(nm) O(nm) 空间复杂度: O ( n 2 ) O(n^2) O(n2)
空间进一步优化,dp数组优化版本
dp[i]只用到前一个物品的状态dp[i-1],dp[j]更新的时候会用到dp[j-v],我们希望用到上一个状态的dp[j-v],而不是更新过后的dp[j-v],简单来说就是dp[j]要先于dp[j-v]更新
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1010;
int n,m,v,w;
int dp[maxn];
int main(){
cin >> n >> m;
for(int i=1;i<=n;i++){
cin >> v >> w;
for(int j=m;j>=v;j--){ // 倒序枚举
// dp[i]只用到前一个物品的状态dp[i-1],dp[j]更新的时候会用到dp[j-v],
// 我们希望用到上一个状态的dp[j-v],而不是更新过后的dp[j-v],
// 简单来说就是dp[j]要先于dp[j-v]更新
dp[j] = max(dp[j],dp[j-v]+w);
}
}
cout << dp[m];
return 0;
}
时间复杂度: O ( n m ) O(nm) O(nm) 空间复杂度: O ( n ) O(n) O(n)
背包常见题目二
如果把题目换成,体积刚好为V时的最大价值,如何求解?
其实跟dp的初始化方式有关,上面用的是把dp全部初始化为0,如果求刚好为V,我们只需按照下面初始化
d p [ 0 ] = 0 d p [ i ] = − I N F ( i ! = 0 ) dp[0]=0 \quad dp[i]=-INF \quad (i!=0) dp[0]=0dp[i]=−INF(i!=0)