混合背包问题

链接: link.
我们使用多重背包的二进制优化，将其分成多个01背包。我们再申请一个bool类型的s数组用来判断我们之后的操作，true为01背包，false为完全背包

#include <bits/stdc++.h>

using namespace std;

const int N = 10010;
int n,m;
int v[N],w[N];
bool s[N];
int dp[N];

int main()
{
    cin>>n>>m;
    int top = 0;
    for(int i=1;i<=n;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        if(!c)//c==0 则是完全背包，对应的s[]中的值为false
        {
            v[top] = a;
            w[top++] = b;
        }
        else
        {
            //多重背包二进制优化
            if(c==-1) c = 1;
            int k = 1;
            while(k<=c)
            {
                v[top] = k*a;
                w[top] = k*b;
                s[top++] = 1;
                c-=k;
                k*=2;
            }
            if(c)
            {
                v[top] = c*a;
                w[top] = c*b;
                s[top++] = 1;
            }
        }
    }
    for(int i=0;i<top;i++)
    {
        if(s[i])//true 01背包
        {
            for(int j=m;j>=v[i];j--)
                dp[j] = max(dp[j],dp[j-v[i]]+w[i]);
        }
        else//false 完全背包
        {
            for(int j=v[i];j<=m;j++)
                dp[j] = max(dp[j],dp[j-v[i]]+w[i]);
        }
    }
    cout<<dp[m]<<endl;
    return 0;
}