题目链接:P1048 采药
01背包问题:
-
状态表示:f[i][j]表示取前i个草药,时间为j的最大价值
-
状态转移:
时间不够用
if(j < v[i]) f[i][j] = f[i - 1][j];
else
时间够用(不选或者选)
f[i][j] = max(f[i - 1][j],f[i - 1][j - v[i]] + w[i];
-
初始化:f[0][0] = 0;
-
res = max(res,f[n][0~m])
代码如下(未优化):
时间、空间复杂度为O(NM)
#include<iostream>
using namespace std;
const int N = 1010;
/*
状态表示:f[i][j]表示取前i个草药,时间为j的最大价值
状态转移:
时间不够用
if(j < v[i]) f[i][j] = f[i - 1][j];
else
时间够用(不选或者选)
f[i][j] = max(f[i - 1][j],f[i - 1][j - v[i]] + w[i];
初始化:f[0][0] = 0;
res = max(res,f[n][0~m])
*/
int m,n;
int w[N],v[N];
int f[N][N];
int res;
int main() {
cin >> m >> n;
for(int i = 1; i <= n; i ++ ){
cin >> v[i] >> w[i];
}
f[0][0] = 0;
for(int i = 1; i <= n; i ++ ){
for(int j = 1; j <= m; j ++ ){
if(j < v[i]){
f[i][j] = f[i - 1][j];
}else{
f[i][j] = max(f[i - 1][j],f[i - 1][j - v[i]] + w[i]);
}
}
}
int res = 0;
for(int i = 1; i <= m; i ++ ){
res = max(res,f[n][i]);
}
cout<<res<<endl;
return 0;
}
优化空间复杂度为O(M):
#include<iostream>
using namespace std;
const int N = 1010;
/*
状态表示:f[i][j]表示取前i个草药,时间为j的最大价值
状态转移:
时间不够用
if(j < v[i]) f[i][j] = f[i - 1][j];
else
时间够用
f[i][j] = max(f[i - 1][j],f[i - 1][j - v[i]] + w[i];
*/
int m,n;
int w[N],v[N];
int f[N];
int main() {
cin >> m >> n;
for(int i = 1; i <= n; i ++ ){
cin >> v[i] >> w[i];
}
for(int i = 1; i <= n; i ++ ){
//因为要用到前i-1个,所以从后往前(保证前面的值不受影响)
for(int j = m; j >= v[i]; j -- ){
f[j] = max(f[j],f[j - v[i]] + w[i]);
}
}
cout<<f[m];
return 0;
}