01背包问题(经典DP)
.
.
南昌理工学院ACM集训队
.
.
01背包问题链接
二维版本
.
我们可以用一个二维数组f[i][j]来存我们的答案,其中一维的f[i]表示的是选了i个物品下的最优解,二维的f[j]表示的是选了体积为j的物品下的最优解,所以f[i][j]的解释就是选了i个物品下j体积的最优解,所以我们的输出的最终结果就是f[n][m]
.
其中f[i][j]就要通过f[i - 1][j]来推导得,例如我们在选第i个物品的时候,我们有两个选择,选这个物品还是不选,不选的话我们保留就是f[i - 1][j]
,选的话结果就是f[i - 1][j - v[i]] + w[i] 。 //当体积为j - v[i]的时候 选了第i个物品,所以用的体积就是j,价值就是f[i - 1][j - v[i]] + w[i]
所以核心代码就是
f[i][j] = max(f[i - 1][j],f[i - 1][j - v[i]] + w[i]);
有了核心代码 这个题就不难了
完整代码
#include <iostream>
using namespace std;
const int N = 1005;
int v[N];
int w[N];
int f[N][N];
int n,m;
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 = 1;j <= m ; j++){
if(j < v[i]){
f[i][j] = f[i- 1][j]; //v[i] < j 选不了 直接等于不选的时候
}else{
f[i][j] = max(f[i - 1][j],f[i - 1][j - v[i]] + w[i]);
}
}
}
cout << f[n][m] << endl;
return 0;
}
一维版本
一维版本就不需要用选了几个物品这一个数组,直接用体积。所以我们定义的时候只用定义f[N]。f[j]表示体积j下的最优解。
核心代码
f[j] = max(f[j],f[j - v[i]] + w[i]);
为什么可以做出这样的变换呢?
例如我们进入循环的时候i = 4,j = 8。v[4] = 4,w[4] = 5。
那么二维的时候有
f[4][8] = max(f[4][8], f[4][4] + 5);
.
.
一维的时候有
f[8] = max(f[8],f[4] + 5);
所以我们要保证f[8]和f[4]都是上一轮的状态,也就是第三轮的状态
所以我们在遍历 j 的时候,需要用到逆序的,如果是顺序的,就不能保证f[8]和f[4]都是上一轮的状态,而逆序的可以保证
完整代码
#include <iostream>
using namespace std;
int n,m;
int f[1005];
int v[1005],w[1005];
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];
}