背包问题总结

都是定义f[i][j]为有前i件物品，j空间下最大价值

1. 0-1背包
   一件物品只能用一次。

f[i][j] = max(f[i - 1][j], f[i - 1][j - v[i]] + w[i]);

代码如下:

for (int i = 1; i <= n; ++i)
    for (int j = 1; j <= totalv; ++j)
        if (j >= v[i])
            f[i][j] = max(f[i - 1][j], f[i - 1][j - v[i]] + w[i]);
        else
            f[i][j] = f[i - 1][j];

时间复杂度 O(n * v) 空间复杂度 O(n * v)

同时可以看到f[i][j]只和f[i - 1][j], f[i - 1][j - v[i]]有关，因此可以做滚动数组优化。
代码如下:

for (int i = 1; i <= n; ++i)
    for (int j = totalv; j >= v[i]; --j)
        f[j] = max(f[j], f[j - v[i]] + w[i]);

时间复杂度 O(n * v) 空间复杂度 O(v)

2. 完全背包
   一件物品可以用多次。

f[i][j] = max(f[i - 1][j], f[i - 1][j - v[i]] + w[i], f[i - 1][j - 2 * v[i]] + w[i] * 2, ...);

代码如下:

for (int i = 1; i <= n; ++i)
    for (int j = 1; j <= totalv; ++j)
        for (int k = 0; k * v[i] <= j; ++k)
            f[i][j] = max(f[i][j], f[i - 1][j - k * v[i]] + k * w[i]);

时间复杂度 O(n * v * v) 空间复杂度 O(n * v)

这里同样可以优化。

f[i][j] = max(f[i - 1][j], f[i - 1][j - v[i]] + w[i], f[i - 1][j - 2 * v[i]] + w[i] * 2, ...);
f[i][j - v[i]] = max(f[i - 1][j - v[i]], f[i - 1][j - 2 * v[i]] + w * w[i], ...);
可以得到
f[i][j] = max(f[i - 1][j], f[i][j - v[i]] + w[i]);

利用滚动数组优化, 最终代码如下:

for (int i = 1; i <= n; ++i)
    for (int j = v[i]; j <= totalv; ++j)
        f[j] = max(f[j], f[j - v[i]] + w[i]);

3. 多重背包
   一件物品最多用指定的次数。

简单地可以用多重背包的未优化解法

f[i][j] = max(f[i - 1][j], f[i - 1][j - v[i]] + w[i], f[i - 1][j - 2 * v[i]] + w[i] * 2, ...);

代码如下:

for (int i = 1; i <= n; ++i)
    for (int j = 0; j <= totalv; ++j)
        for (int k = 0; k <= cnt[i] && k * v[i] <= j; ++k)
            f[i][j] = max(f[i][j], f[i - 1][j - k * v[i]] + w[i] * k);

多重背包可以使用二进制优化，即如某件物品体积为v, 价值为w， 最多用s次。
可以将其拆成0-1背包，当s = 6时，可以拆成v + 2v + 3v;

代码如下：

const int N = 30010;
int f[N], w[N], v[N];
int cnt = 1, n, m;

int main() {
    cin >> n >> m;
    int vv, ww, ss;
    for (int i = 0; i < n; ++i) {
        cin >> vv >> ww >> ss;
        int k = 1;
        while (k <= ss) {
            w[cnt] = k * ww;
            v[cnt] = k * vv;
            ss -= k;
            k *= 2;
            ++cnt;
        }
        if (ss) {
            w[cnt] = ss * ww;
            v[cnt] = ss * vv;
            ++cnt;
        }
    }
    
    for (int i = 1; i < cnt; ++i)
        for (int j = m; j >= v[i]; --j)
            f[j] = max(f[j], f[j - v[i]] + w[i]);
    
    cout << f[m];
    return 0;
}

4. 分组背包
   每个组有若干物品，且一个组只能选择一个，和多重背包差不多思路，代码如下：

f[i][j] = max(f[i - 1][j], f[i-1][j - v0] + w0, f[i - 1][j - v1] + w1,...)

for (int i = 1; i <= n; ++i)
    for (int j = m; j >= 0; --j)
        for (int k = 0; k < gnum[i]; ++k)
            if (j >= v[i][k])
                f[j] = max(f[j], f[j - v[i][k]] + w[i][k]);