混合背包模板(01+多重+完全)

描述

---

 

输入w,v,m.m表示数量,m=-1表示无限多个.

题解

---

 

 

#include<cstdio>
#include<algorithm>
using std :: max;

const int maxn=205,maxw=200005;
int dp[maxw],v[maxn],w[maxn],m[maxn];
int N,W;

void zero_one(int vi,int wi) { for(int j=W;j>=wi;j--) dp[j]=max(dp[j],dp[j-wi]+vi); }

void complete(int vi,int wi) { for(int j=wi;j<=W;j++) dp[j]=max(dp[j],dp[j-wi]+vi); } 

void multy(int i,int vi,int wi,int mi)
{
    for(int k=1;k<mi;k*=2)
    {
        zero_one(k*vi,k*wi);
        mi-=k;
    }
    zero_one(mi*vi,mi*wi);
}

void solve()
{
    for(int i=1;i<=N;i++)
    {
        if(m[i]==-1) complete(v[i],w[i]);
        else if(m[i]==1) zero_one(v[i],w[i]);
        else multy(i,v[i],w[i],m[i]);
    }
    printf("%d
",dp[W]);
}

void init()
{
    scanf("%d%d",&N,&W);
    for(int i=1;i<=N;i++) scanf("%d%d%d",&w[i],&v[i],&m[i]);
}

int main()
{
    init();
    solve();
    return 0;
}