动态规划学习记录
二维dp数组01背包
使用二维dp数组dp[i][j],dp[i][j] 表示从下标为[0-i]的物品里任意取,放进容量为j的背包,价值总和最大是多少。
那么可以有两个方向推出来dp[i][j]:放物品i,或者不放物品i
1.由dp[i - 1][j]推出,即背包容量为j,里面不放物品i的最大价值,此时dp[i][j]就是dp[i - 1][j]
2.由dp[i - 1][j - weight[i]]推出,dp[i - 1][j - weight[i]] 为背包容量为j - weight[i]的时候不放物品i的最大价值,那么dp[i - 1][j - weight[i]] + value[i] (物品i的价值),就是背包放物品i得到的最大价值
二者的最大值就是dp[i][j]
所以递归公式:dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);
由状态转移方程 可以看出i 是由 i-1 推导出来,那么i为0的时候就一定要初始化。
dp[0][j],即:i为0,存放编号0的物品的时候,各个容量的背包所能存放的最大价值。
代码如下:
// 倒叙遍历
for (int j = bagWeight; j >= weight[0]; j--) {
dp[0][j] = dp[0][j - weight[0]] + value[0]; // 初始化i为0时候的情况
}
大家应该发现,这个初始化为什么是倒叙的遍历的?正序遍历就不行么?
正序遍历还真就不行,dp[0][j]表示容量为j的背包存放物品0时候的最大价值,物品0的价值就是15,因为题目中说了每个物品只有一个!所以dp[0][j]如果不是初始值的话,就应该都是物品0的价值,也就是15。
但如果一旦正序遍历了,那么物品0就会被重复加入多次!例如代码如下:
// 正序遍历
for (int j = weight[0]; j <= bagWeight; j++) {
dp[0][j] = dp[0][j - weight[0]] + value[0];
}
例如dp[0][1] 是15,到了dp[0][2] = dp[0][2 - 1] + 15; 也就是dp[0][2] = 30 了,那么就是物品0被重复放入了。
遍历代码
先遍历物品,然后遍历背包重量的代码。
// weight数组的大小 就是物品个数
for(int i = 1; i < weight.size(); i++) { // 遍历物品
for(int j = 0; j <= bagWeight; j++) { // 遍历背包容量
if (j < weight[i]) dp[i][j] = dp[i - 1][j]; // 这个是为了展现dp数组里元素的变化
else dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);
}
}
例题以及代码:例题链接.
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int v[25],p[25];
int dp[25][30000] = {0};
int main()
{
int n,m;
cin>>n>>m;
cout<<dp[4][1000]<<endl;;
for(int i= 0;i<m;i++)
{
cin>>v[i]>>p[i];
}
for(int i = n;i>= 0;i--)
{
if(i>v[0])//初始化要保证数组不越界
dp[0][i] = dp[0][i-v[0]]+p[0]*v[0];
}
for(int i = 1;i<m;i++)
{
for(int j = 0;j<=n;j++)
{
if(j<v[i]) dp[i][j] = dp[i-1][j];
else dp[i][j] = max(dp[i-1][j],dp[i-1][j-v[i]]+v[i]*p[i]);
}
}
cout<<dp[4][1000];
return 0;
}