朴素多重背包解法:
#include <iostream>
using namespace std;
const int N = 1010;
int v[N], w[N], s[N];
int f[N][N];
int main()
{
int n, m;
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] & j - k * v[i] >= 0; k ++)//关于为何k从0开始
//当k为0时,后半部分恰好为f[i - 1][j],将不选该物品的情况包含在内
{
f[i][j] = max(f[i][j], f[i - 1][j - k * v[i]] + k * w[i]);
}
}
}
cout << f[n][m] << endl;
return 0;
}
我们可以发现朴素解法时间复杂度较高,因此可使用二进制优化原算法:
#include<iostream>
using namespace std;
const int N=11010,M=2010;
int n,m;
int v[N],w[N];
int f[N];
int main()
{
cin >> n >> m;
int cnt = 1;
for(int i = 1; i <= n; i ++ )//拆分打包
{
int a,b,s;
cin >> a >> b >> s;
int k = 1;
while(k <= s)//分别打包成1,2,4,8。。。。
{
v[cnt] = a * k;
w[cnt] = b * k;
s -= k;
k *= 2;
cnt ++;
}
if(s > 0)
{
v[cnt] = s * a;
w[cnt] = s * b;
cnt ++;
}
}
for(int i = 1; i <= cnt; i ++ )
{
for(int j = m; j >= v[i]; j -- )//使用01背包的一维法进一步优化空间复杂度
{
f[j] = max(f[j], f[j - v[i]] + w[i]);
}
}
cout << f[m] << endl;
return 0;
}