题目地址:
https://www.acwing.com/problem/content/3/
有 N N N种物品和一个容量是 V V V的背包,每种物品都有无限件可用。第 i i i种物品的体积是 v i v_i vi,价值是 w i w_i wi。求解将哪些物品装入背包,可使这些物品的总体积不超过背包容量,且总价值最大。输出最大价值。
数据范围:
0 ≤ N , V ≤ 1000 0\le N, V\le 1000 0≤N,V≤1000
0 ≤ v i , w i ≤ 1000 0\le v_i,w_i\le 1000 0≤vi,wi≤1000
思路是动态规划。设 f [ i ] [ j ] f[i][j] f[i][j]是只允许选前 i i i个物品,并且总重量不超过 j j j的情况下,能得到的最大价值。那么所有方案可以分为两类,不含第 i i i个物品和含第 i i i个物品,若不含第 i i i个物品,那就相当于只从前 i − 1 i-1 i−1个物品,最大价值就是 f [ i − 1 ] [ j ] f[i-1][j] f[i−1][j];若含第 i i i个物品,最大价值就是 f [ i ] [ j − v i ] + w i f[i][j-v_i]+w_i f[i][j−vi]+wi。则有: f [ i ] [ j ] = min { f [ i − 1 ] [ j ] , f [ i ] [ j − v i ] + w i } f[i][j]=\min\{f[i-1][j],f[i][j-v_i]+w_i\} f[i][j]=min{ f[i−1][j],f[i][j−vi]+wi}我们可以优化空间。可以发现 f [ i ] [ j ] f[i][j] f[i][j]只会用到其上和左边的值,所以可以只开一行数组,然后每轮都从左向右更新。代码如下:
#include <iostream>
using namespace std;
const int N = 1010, V = 1010;
int v[N], w[N];
int f[V];
int main() {
int n, m;
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[j] = max(f[j], f[j - v[i]] + w[i]);
cout << f[m] << endl;
return 0;
}
时间复杂度 O ( N V ) O(NV) O(NV),空间 O ( V ) O(V) O(V)。