多重背包
有N种物品,第i种物品的体积为Ci,价值是Wi,但是每种物品的数量都是有限的,为ni。现在有容量为V的背包,请你放入若干物品,使获得的价值尽量大。
朴素算法:
把N种物品逐个拆分,得到Σni个物品,则原问题可转化为01背包求解。这样做的时间复杂度为O(V×Σn)。
或者是在枚举种类的过程中枚举个数k,但时间复杂度也是O(V×Σn)。
二进制优化
我们可以考虑二进制的思想,将第i种物品通过二进制拆分成若干件物品,
比如20件物品可以用1,2,4,8,5这5个数进行组合并相加,就能得到20以内的任何数。每一个ni优化到log ni
代码
#include<iostream>
#include<cstring>
#define max(a, b) (a) > (b) ? (a) : (b)
using namespace std;
int dp[500000], v[1005], w[1005], n[1005];
int N, M;
//多重背包二进制优化
//---------------------------------------------------------------------------
//01背包
void ZeroPack(int v, int w){
for(int j = M; j >= w; j--){
dp[j] = max(dp[j], dp[j - w] + v);
}
return ;
}
//完全背包,每件物品可以选择无数件
void CompletePack(int v, int w){
for(int i = w; i <= M; i++){
dp[i] = max(dp[i], dp[i - w] + v);
}
return ;
}
//多重背包二进制优化
void erjinzhi(){
int k, m;
memset(dp, 0, sizeof(dp));
for(int i = 1; i <= N; i++){
if(M <= n[i] * w[i]){ //如果总的容量比这个物品所有的还要多,完全背包
//cout << "11111" << endl;
CompletePack(v[i], w[i]);
//cout << "2222" << endl;
}else{ //否则转换为01背包
for(k = 1; k * 2 < n[i]; k = k * 2){//log(ni)
ZeroPack(k * v[i], k * w[i]);
}
m = n[i] - k + 1;//最后还剩下m个
ZeroPack(m * v[i], m * w[i]);
}
}
return ;
}
//-------------------------------------------------------------------------------------
int main(){
while(cin >> M >> N){
for(int i = 1; i <= N; i++){
cin >> n[i] >> v[i];
w[i] = v[i];
}
erjinzhi();
cout << dp[M] << endl;
}
return 0;
}