上一次我们对01背包和完全背包进行了探究,今天,我们将孑孓解决多重背包和混合三种背包!
1.多重背包
多重背包是指,有n种物品,每一种物品有n[i]件,每一种物品也有自己的价值和重量。现在有一个容积为m的背包,求能装下的最大价值。
拿到这一题,首先分析如何列状态,如果用f[i][j]表示前i种物品放入不超过j体积的最大价值,那么f[i][j]=max{f[i-1][j],f[i-1][j-k*w[i]]+c[i]*k}(0<=k<=n[i]),为什么呢?大家想想,01背包的思想就是放一个或不放,那么多重背包的思想就是放0个,放一个,放两个……,所以我们可以多用一重循环来写决策,程序如下:
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
using namespace std;
int w[2001],c[2001],s[2001],f[2001][2001];
int main(){
int i,j,k,n,m;
cin>>n>>m;
for(i=1;i<=n;i++)
cin>>w[i]>>c[i]>>s[i];
for(i=1;i<=n;i++)
for(j=m;j>=0;j--)
for(k=0;k<=s[i];k++)
if(k*w[i]>j)break;
else f[i][j]=max(f[i][j],f[i-1][j-w[i]*k]+k*c[i]);
cout<<f[n][m]<<endl;
return 0;
}
但是这种方法还可以有改进,把数组改成一维的滚动数组,如下代码:
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
using namespace std;
int w[2001],c[2001],s[2001],f[2001];
int main(){
int i,j,k,n,m;
cin>>n>>m;
for(i=1;i<=n;i++)
cin>>w[i]>>c[i]>>s[i];
for(i=1;i<=n;i++)
for(j=m;j>=0;j--)
for(k=0;k<=s[i];k++)
if(k*w[i]>j)break;
else f[j]=max(f[j],f[j-w[i]*k]+k*c[i]);
cout<<f[m]<<endl;
return 0;
}
拓展:二进制求多重背包。
上篇文章完全背包的拓展中,已经讲述了如何将完全背包转换成01背包,那么多重背包又怎么办呢?
完全背包代码
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
using namespace std;
int w[100001],c[100001],f[100001];
int main(){
int i,j,k,n,m,a,b;
cin>>m>>n;
int n1=0;
for(i=1;i<=n;i++){
cin>>a>>b;
k=m/b;
int t=1;
while(k>=t){
w[++n1]=a*t;c[n1]=b*t;
k-=t;t*=2;
}
c[++n1]=b*k;w[n1]=a*k;
}
for(i=1;i<=n1;i++)
for(j=m;j>=w[i];j--)
f[j]=max(f[j],f[j-w[i]]+c[i]);
cout<<f[m]<<endl;
return 0;
}
多重背包代码
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
using namespace std;
int w[100001],c[100001],f[100001];
int main(){
int i,j,k,n,m,a,b;
cin>>n>>m;
int n1=0;
for(i=1;i<=n;i++){
int s;
cin>>a>>b>>s;k=s;
int t=1;
while(k>=t){
w[++n1]=a*t;c[n1]=b*t;
k-=t;t*=2;
}
c[++n1]=b*k;w[n1]=a*k;
}
for(i=1;i<=n1;i++)
for(j=m;j>=w[i];j--)
f[j]=max(f[j],f[j-w[i]]+c[i]);
cout<<f[m]<<endl;
return 0;
}
2.混合三种背包的问题
这种类型的题目看上去把很多类型的题目混合在一起,01、完全、多重背包,但是实际上只需要加一个判断就好,如果总数为0,那么就是完全背包,否则按照多重背包解。那么直接看代码:
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
using namespace std;
int w[100001],c[100001],s[100001],f[100001];
int main(){
int i,j,k,n,m;
cin>>m>>n;
for(i=1;i<=n;i++)
cin>>w[i]>>c[i]>>s[i];
for(i=1;i<=n;i++)
if(s[i]==0){
for(j=w[i];j<=m;j++)
f[j]=max(f[j],f[j-w[i]]+c[i]);
}
else{
for(k=0;k<=s[i];k++)
for(j=m;j>=w[i];j--)
f[j]=max(f[j],f[j-w[i]]+c[i]);
}
cout<<f[m]<<endl;
return 0;
}