混合背包模板(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\n",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;
}