01背包问题:有n件物品和一个最多能背重量为w 的背包。第i件物品的重量是weight[i],得到的价值是value[i] 。每件物品只能用一次,求解将哪些物品装入背包里物品价值总和最大。
dp数组以及下标的含义:dp[i][j]表示从下标0到i的物品里任意选,放进容量为j的背包里,价值总和最大是多少。
dp递推公式:dp[i][j]取决于放不放物品i。
不放物品i:dp[i][j] 由dp[i - 1][j]推出,即背包容量为j,里面不放物品i,从下标0到i - 1 的物品里任意选得到的最大价值,此时dp[i][j]就是dp[i - 1][j]。
放物品i:由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得到的最大价值。
初始化:i = 0时,即dp[0][j]的初始化,当j < weigit[0]时,dp[0][j] = 0; j >= weight[0],dp[0][j] = weight[0]。j = 0时,即背包能装的重量为0,dp[i][0] = 0;
遍历顺序:dp[i][j]的状态是由上一行和左上角的dp值确定,因此先遍历i即物品或者先遍历背包j都行。
具体代码:
#include <iostream>
#include <vector>
#include <math.h>
int test(){
vector<int> weight = { 1, 3, 4 };
vector<int> value = { 15, 20, 30 };
int bagweight = 4;
//dp[i][j]表示从下标0到i的物品中任意选取放进容量为j的背包中所能获得的最大价值是多少
vector<vector<int>> dp(weight.size(), vector<int>(bagweight + 1, 0));
//初始化dp[0][j]
for (int j = weight[0]; j <= bagweight; j++) {
dp[0][j] = value[0];
}
//初始化dp[i][0]
for (int i = 0; i < weight.size(); i++) {
dp[i][0] = 0;
}
//遍历
for (int i = 1; i < weight.size(); i++) {
for (int j = 0; j <= bagweight; j++) {
/*
*当前背包的容量都没有当前物品i大的时候,是不放物品i的
* 那么前i - 1个物品能放下的最大价值就是当前情况的最大价值
*/
if (j < weight[i]) dp[i][j] = dp[i - 1][j];
/*
* 当前背包的容量可以放下物品i
* 那么此时分两种情况:
* 1、不放物品i
* 2、放物品i
* 比较这两种情况下,哪种背包中物品的最大价值最大
*/
else dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);
//cout << dp[i][j];
}
//cout << endl;
}
return dp[weight.size() - 1][bagweight];
}
int main() {
int max = test();
cout << max << endl;
//test();
return 0;
}