Acwing算法复习之01背包问题

先把例题（模板题）列出来，然后再解析。

题目：（来源acwing）

有 N 件物品和一个容量是 V 的背包。每件物品只能使用一次。

第 i 件物品的体积是 vi，价值是 wi。

求解将哪些物品装入背包，可使这些物品的总体积不超过背包容量，且总价值最大。
输出最大价值。

输入格式

第一行两个整数，N，V，用空格隔开，分别表示物品数量和背包容积。

接下来有 N 行，每行两个整数 vi,wi，用空格隔开，分别表示第 i 件物品的体积和价值。

输出格式

输出一个整数，表示最大价值。

数据范围

0  <  N,  V  ≤  10000  <  N,  V  ≤  1000
0  <  vi, wi  ≤  10000  <  vi,  wi  ≤  1000

输入样例

4 5
1 2
2 4
3 4
4 5

输出样例：

8

这是01背包问题的模板题，我们就用这个题来讲解一下。

01背包问题本质是动态规划问题。那我们就要从状态表示和状态计算（也就是状态转移方程）来考虑问题。

方法一：二维表示

（1）用二维数组 f [ N ] [ N ]  来表示， 前 i 个物品， 背包容积为 j 的情况下， 背包所能放入物品的价值最大值。

（2）得知状态表示后， 我们需要考虑状态计算。 也就是 每个 f [ i ] [ j ] 怎么计算出来。

那么这时候就要考虑此时背包的体积 j 和 第 i 个物品的体积间的关系了。

          1> 当 背包体积 j < 第 i 个物品体积 v [ i ] 时， 我们肯定不能选第 i 个物品，因为背包放不下了， 所以 f [ i ] [ j ] = f[ i - 1] [ j ]。

          2>当 背包体积 j >= 第 i 个物品体积 v [ i ] 时， 我们就既可以选择第  i 个物品， 也可以选择不选。如果 不选第 i 个物品，那 f [ i ] [ j ]  =  f[ i - 1 ] [ j ]。如果 选 第 i 个物品，那f [ 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 ] )。最后 f [ n ] [ m ] 就是目标答案。

#include <bits/stdc++.h>

using namespace std;

const int N = 1010;

int w[N], v[N], f[N][N];

int main()
{
    int n, m;
    cin >> n >> m;
    
    for (int i = 1; i <= n; i ++) scanf("%d%d", &v[i], &w[i]);
    
    for (int i = 1; i <= n; i ++)
        for (int j = 1; j <= m; j ++)
        {
            f[i][j] = f[i - 1][j];
            if (j >= v[i]) f[i][j] = max(f[i][j], f[i - 1][j - v[i]] + w[i]);
        }
        
    cout << f[n][m] << endl;
    
    return 0;
}

方法二：一维表示

前面我们用二维数组 f [ N ] [ N ] 来表示 任意  i 和 任意 j 状态下 背包所能存储价值最大值。但我们要求的只是在给定最大体积m 时背包能存储的价值最大值。那么我们可以用一维数组 f [ N ] 来表示背包体积为 j 的情况 背包所能存储的价值最大值。

     那么我们来考虑状态计算，当外层循环到 第 i 个物品， 内层循环到 体积 j 时，f [ j ] 的 取值有两种方式。

     1 > 不取第 i 个物品，也就是第 i 个物品不占据背包体积，f [ j ] = f[ j ]。

     2 > 取第  i 个物品，也就是当背包体积 j >= v [ i ] 时， f [ j ] = f [ j - v[ i ] ] + w[ i ] 。

所以取最大值，f [ j ] = max ( f [ j ], f [ j - v[ i ] ] + w[ i ] )。

好像到这位置已经结束了， 但有一点我们不能忽略了。那就是与二维数组循环不同，题目要求每个物品只能使用一次，那么在一维做法下，外层循环到第 i 个物品， 内层循环到体积 j ，当我们打算把第 i 个物品放入背包时， f [ j ] = f[ j - v[ i ] ] + w[ i ] 。请注意此时我们用 f [ j - v[ i ] ] 来更新    f [ j ] ,如果我们照搬二维循环方式，也就是体积从 1 到 最大 m（从小到大）的话，f [ j - v [ i ] ]先于 f [ j ]计算， 极有可能 f [ j - v [ i ] ] 已经把第 i 个物品放入背包内，也就是说可能会把第 i 个物品重复放入背包了，这就不符合01背包要求了。所以我们要反过来，体积从大到小，这样就不会重复使用了。代码如下。

#include <bits/stdc++.h>

using namespace std;

const int N = 1010;

int v[N], w[N], f[N];

int main()
{
    int n, m;
    cin >> n >> m;
    
    for (int i = 1; i <= n; i ++) scanf("%d%d", &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] << endl;
    
    return 0;
}