多重背包问题

朴素代码

题面

样例

思路

跟完全背包思路一样，三重循环，但是在最内层循环有一个判断条件,k<=s[i](个数不能多于总个数)，f[i][j]=max(f[i][j],f[i-1][j-k*v[i]]+k*w[i])

代码

#include<iostream>

using namespace std;

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

int main(void){
    cin >> n >> m;
    
    for(int i=1;i<=n;i++) cin >> v[i] >> w[i] >> s[i];
    
    for(int i=1;i<=n;i++){
        for(int j=0;j<=m;j++){
            for(int k=0;k<=s[i]&&v[i]*k<=j;k++)
            f[i][j] = max(f[i][j],f[i-1][j-v[i]*k]+w[i]*k);
        }
    }
    
    cout << f[n][m];
    
    return 0;
}

优化代码

题面

样例

优化思路

我们回忆在完全背包优化的时候是将f[i][j]展开，然后后k项可以用一项表示出来，但是这里我们不行了，为什么呢？请看下图

f[i][j-v]这一项最后面那一项多出了，与前面f[i][j]构不成等大关系，所以此种优化思路不可行。

但是这里我们采用一种全新的优化方式--二进制优化

二进制优化

给定一个任意的数s，我们可以写出其二进制形式，比如58 = 111010 = （32+16+8+2），但是我们的32不可以要，因为如果加上了32，那么它的范围将会0-63，但是我们只要到58，所以我们要从后往前去表示(从二进制的末尾开始)，就是011111 = （1+2+4+8+16）=31，58-31=27，我们另一个新的c=27(也可以理解为原来哪一位32换成了27，但是代码中我们不会这样做)，那么我们就可以刚好表示0-58之间所有的数字不多不少刚好。

此时我们的时间复杂度从原来的N*M*S->N*M*log(S)

代码

#include<iostream>

using namespace std;

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

int main(void){
    cin >> n >> m;
    int cnt=0;
    for(int i=1;i<=n;i++){
        int a,b,s;
        cin >> a >> b >> s;
        int k=1;
        while(k<=s){
            cnt++;
            v[cnt] = a*k;
            w[cnt] = b*k;
            s-=k;
            k*=2;
        }
        if(s>0){
            cnt++;
            v[cnt] = a*s;
            w[cnt] = b*s;
        }
    }
    
    n=cnt;
    
    // 做一次01背包问题
    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;
}