背包问题——7. 混合背包

混合背包问题

题目描述

有 N 种物品和一个容量是 V 的背包。物品一共有三类，每种体积是 vi，价值是 wi。

第一类物品只能用1次（01背包）；
第二类物品可以用无限次（完全背包）；
第三类物品最多只能用 si 次（多重背包）；

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

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

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

si=−1 表示第 i 种物品只能用1次；
si=0 表示第 i 种物品可以用无限次；
si>0 表示第 i 种物品可以使用 si 次；

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

数据范围
0<N,V≤1000
0<vi,wi≤1000
−1≤si≤1000

输入样例:

4 5
1 2 -1
2 4 1
3 4 0
4 5 2

输出样例：

8

分析：
原来的二维dp表每一行的物品都是同样类型的物品，可以用同样的处理手段，
现在每一行的物品可能是不一样的物品，因此对每一行进行更新的时候，先判断它的物品类型，再用相应的方法进行求解，更新该行。
主要可以分为两种： 完全背包和多重背包，01背包就是物品使用次数为1的多重背包。

分析不同情况，进行状态转移即可，如下图：

代码实现

#include <iostream>
#define read(x) scanf("%d",&x)

using namespace std;

const int maxn=1010,maxv=1010;
int v[maxn],w[maxn],s[maxn];
int dp[maxv];

int main() {
    int N,V;
    read(N),read(V);
    for (int i=1;i<=N;i++) read(v[i]),read(w[i]),read(s[i]);

    for (int i=1;i<=N;i++) {
        if (s[i]==0) { //完全背包
            for (int j=v[i];j<=V;j++)  dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
        } else {
            if (s[i]==-1) s[i]=1; //0-1背包就是特殊的多重背包
            for (int k=1;k<=s[i];k*=2) {//每个背包挨个拆分
                for (int j=V;j>=k*v[i];j--)  dp[j]=max(dp[j],dp[j-k*v[i]]+k*w[i]);
                s[i]-=k;
            }
            if(s[i]) {
                for (int j=V;j>=s[i]*v[i];j--)  dp[j]=max(dp[j],dp[j-s[i]*v[i]]+s[i]*w[i]);
            }
        }
    }
    
    printf("%d",dp[V]);

    return 0;
}

也可以节省内存，读入一次，处理一次，一次处理一行。

#include <iostream>

using namespace std;

const int N = 1010;

int n, m;
int f[N];

int main() {

    cin >> n >> m;
    for (int i = 0; i < n; i++) {
        int v, w, s;
        cin >> v >> w >> s;
        if (s == 0) {  // 完全背包
            for (int j = v; j <= m; j++) f[j] = max(f[j], f[j - v] + w);
        } else {
            if (s == -1) s = 1;  // 全部转为多重背包，然后使用二进制优化
            for (int k = 1; k <= s; k *= 2) {
                for (int j = m; j >= k * v; j--)
                    f[j] = max(f[j], f[j - k * v] + k * w);
                s -= k;
            }
            if (s) {
                for (int j = m; j >= s * v; j--)
                    f[j] = max(f[j], f[j - s * v] + s * w);
            }
        }
    }
    cout << f[m] << endl;
    return 0;
}