链接: 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;
}