有N
件物品和一个容量为V
的背包,第i
件物品的费用是c[i]
,价值是w[i]
,求将哪些物品装入背包可使总费用不超过背包的容量,且价值之和最大。
重点:每件物品仅有一件,可以取或不取。
解题思路:
定义F[i][v]
表示将第i
件物品放入一个容量为v
的背包的价值,则其状态转移方程如下:
F[i][v] = max{F[i-1][v], F[i-1][v-c[i]] + w[i]}
上面的这个方程的含义如下:
i
件物品放入容量为v
的背包的最大价值,只需要考虑第i
件物品是否放入背包,有放和不放两种情况。i
件物品,那么总价值就是前i-1
件物品放入容量为v
的背包的价值,也就是F[i-1][v]
。i
件物品,那么就价值就是第i
件物品的价值w[i]
,加上前i-1
件物品放入容量为v-c[i]
的背包的价值F[i-1][v-c[i]]
。i
件物品放入容量为v
的背包的最大价值。示例代码实现如下:
#define MAXN 1005 int n; // 物品数 int v; // 背包容量 int c[MAXN]; // 物品费用 int w[MAXN]; // 物品价值 int f[MAXN][MAXN]; // f[i][j],j体积下前i个物品的最大价值 int knapsack01() { cin >> n >> v; for(int i = 1; i <= n; i++) { // 下标从1开始 cin >> c[i] >> w[i]; } // 注意这里隐含设置了取0件物品和背包容量为0时f数组对应的值全为0的情况 // for(int i = 0; i <= v; i++) f[0][i] = 0; // for(int i = 0; i <= n; i++) f[i][0] = 0; for(int i = 1; i <= n; i++) { for(int j = 1; j <= v; j++) { if(j < c[i]) { // 当前背包容量j装不下第i个物品,则价值等于前i-1个物品 f[i][j] = f[i-1][j]; } else { // 能装,按方程进行决策 // 注意,当到j恰好能装下全部的前i件物品时, // 后续的j可直接设置成f[i][j-1],因为后续的j表示背包容量已经有冗余了 f[i][j] = max(f[i-1][j], f[i-1][j - c[i]] + w[i]); } } } return f[n][v]; } |
优化空间复杂度:
根据状态转移方程可知,第i
件物品的计算过程只与第i-1
件物品有关,由于可以想到使用滚动数组的方式,将空间复杂度降为从二维降至一维。
f[j]
,表示N
件物品在背包容量为j
下的最优解。j
必须从V
枚举到0
,也就是逆序。为什么采用逆序,参考以下解释:
dp[j] = max(dp[j], dp[j - v[i]] + w[i]) 假如枚举到:i = 3, j = 8, v[3] = 5, w[3] = 1 二维:dp[3][8] = max(dp[2][8], dp[2][3] + w[3]) 此时的dp[2][8]和dp[2][3]都是上一轮的状态值 一维:dp[8] = max(dp[8], dp[3] + w[3]) 我们要保证dp[8]和dp[3]都是上一轮的状态值 按照逆序的顺序,一维dp数组的更新顺序为:dp[8], dp[7], dp[6], ... , dp[3] 也就是说,在本轮更新的值,不会影响本轮中其他未更新的值!较小的index对应的状态是上一轮的状态值! 如果按照顺序进行更新,dp[3] = max(dp[3], dp[0] + w[0]),对dp[3]的状态进行了更新,那么在更新dp[8]时,用到的dp[3] 就不是上一轮的状态了,不满足动态规划的要求。 作者:polaris 链接:https://www.acwing.com/solution/content/3982/ 来源:AcWing 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。 |