混合背包——二进制优化

 

输入样例

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

输出样例：

8

#include <bits/stdc++.h>
using namespace std;

const int N = 10010;
int n, m;
int f[N];
int v[N], w[N];

int main(){
    int a, b, s;
    cin >> n >> m;
    int cnt = 0;
    while(n--){
        cin >> a >> b >> s;
        if(s == -1){//第一种情况
            v[cnt] = a; w[cnt] = b;
            cnt++;
        }
        else if(s > 0){//第三种情况
            for(int i = 1; i <= s; i *= 2){
                v[cnt] = i * a; w[cnt] = i * b;
                s -= i;
                cnt++;
            }
            if(s > 0){//剩余的s
                v[cnt] = s * a; w[cnt] = s * b;
                cnt++;
            }
        }
        else if(s == 0){//第二种情况
            for(int i = 1; i <= N; i *= 2){//不限个数，使用最大的情况
                v[cnt] = i * a; w[cnt] = i * b;
                cnt++;
            }
        }

    }
    n = cnt;//更新n
    for(int i = 0; i <= n; i++) //0-1背包
        for(int j = m; j >= v[i]; j--){
            f[j] = max(f[j - v[i]] + w[i], f[j]);
        }
    cout << f[m];
    return 0;
}