【提高】混合背包c++

描述

有 N 种物品和一个容量是 V 的背包。

物品一共有三类：
第一类物品只能用1次（01背包）；
第二类物品可以用无限次（完全背包）；
第三类物品最多只能用 si 次（多重背包）；
每种体积是 vi，价值是 wi。

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

输入描述

第一行两个整数，N，V，用空格隔开，分别表示物品种数和背包容积。
接下来有 N 行，每行三个整数 vi,wi,si，用空格隔开，分别表示第 i 种物品的体积、价值和数量。
si=−1 表示第 i 种物品只能用1次；
si=0 表示第 i 种物品可以用无限次；
si>0 表示第 i 种物品可以使用 si 次；

输出描述

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

用例输入 1 

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

用例输出 1 

8

提示

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

来源

背包问题 动态规划

代码

#include <bits/stdc++.h>
using namespace std;
int n, m, cnt;
long long f[20000005];
long long w[20000005], v[20000005];
int main() {
    cin >> n >> m;
    cnt = 0;
    for (int sum, x, y, a, i = 1; i <= n; i++) {
        cin >> x >> y >> sum;
        if (sum != -1) {
            if (sum == 0)
                sum = m / x + 1;
            a = 1;
            while (a <= sum) {
                cnt++;
                w[cnt] = x * a, v[cnt] = y * a;
                sum -= a, a *= 2;
            }
            if (sum > 0) {
                cnt++;
                w[cnt] = x * sum, v[cnt] = y * sum;
            }
        } else {
            cnt++;
            w[cnt] = x, v[cnt] = y;
        }
    }
    for (int i = 1; i <= cnt; i++) {
        for (int j = m; j >= w[i]; j--) {
            f[j] = max(f[j], f[j - w[i]] + v[i]);
        }
    }
    cout << f[m];
    return 0;
}