今天学习的是多重背包问题😎
其实昨天晚上就开始看了,但是实在太懒了就没学完
先放题:
急!灾区的食物依然短缺!
为了挽救灾区同胞的生命,心系灾区同胞的你准备自己采购一些粮食支援灾区,现在假设你一共有资金n元,而市场有m种大米,每种大米都是袋装产品,其价格不等,并且只能整袋购买。
请问:你用有限的资金最多能采购多少公斤粮食呢?
后记:
人生是一个充满了变数的生命过程,天灾、人祸、病痛是我们生命历程中不可预知的威胁。
月有阴晴圆缺,人有旦夕祸福,未来对于我们而言是一个未知数。那么,我们要做的就应该是珍惜现在,感恩生活——
感谢父母,他们给予我们生命,抚养我们成人;
感谢老师,他们授给我们知识,教我们做人
感谢朋友,他们让我们感受到世界的温暖;
感谢对手,他们令我们不断进取、努力。
同样,我们也要感谢痛苦与艰辛带给我们的财富~
Input
输入数据首先包含一个正整数C,表示有C组测试用例,每组测试用例的第一行是两个整数n和m(1<=n<=100, 1<=m<=100),分别表示经费的金额和大米的种类,然后是m行数据,每行包含3个数p,h和c(1<=p<=20,1<=h<=200,1<=c<=20),分别表示每袋的价格、每袋的重量以及对应种类大米的袋数。
Output
对于每组测试数据,请输出能够购买大米的最多重量,你可以假设经费买不光所有的大米,并且经费你可以不用完。每个实例的输出占一行。
Sample Input
1
8 2
2 100 4
4 100 2
Sample Output
400
自己写了一下,例题过了,但还是WA了,这是初始代码:
/**
* TODO Auther: by SR
* Date: 2020-01-23 22:40:19
* LastEditTime: 2020-01-24 11:02:19
* ! 今日运势: 吉,无bug
* ?多重背包问题
* *求最大值
**/
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<string>
#include<memory.h>
using namespace std;
struct rice{
int p,h,c;
}a[102];
void multiple_bag();
int ans[102];
int C,money,m;
int main(){
cin >> C;
while(C--){
ans[0] = 0;
memset(ans,0,sizeof(ans));
cin >> money >> m;
for(int i = 1; i <= m; i++){
cin>>a[i].p>>a[i].h>>a[i].c;
}
multiple_bag();
}
return 0;
}
void multiple_bag(){
int temp = money;
for(int i = 1; i <= m; i++){
for(int j = a[i].p;j <= 2*a[i].p; j++){
int temp_c = a[i].c;
for(int k = j; k <= money && temp_c > 0; k += a[i].p, temp_c--){
ans[k] = max(ans[k], ans[k-a[i].p] + a[i].h);
}
}
}
int large = 0;
cout << ans[money] <<"\n";
return;
}
基本思路应该没问题,就是在多重背包的基础上加上数量限制。应该还是实现过程中出现的问题。
查了一下,朴素的解决方法是直接用k个空间存储数量为k的大米袋数,然后就转化为01背包了。但这样的空间复杂度是(N2)内存,时间复杂度是(N3),也很容易爆时间,所以要进行优化,于是我又查到了一个叫二进制拆分的思路,就不解释了,有兴趣自己查。总之优化后的空间复杂度是(N*logN),时间复杂度是(N2*logN)。好像还是很大。。
先实现了再说:
OJ炸了,现在交不了☹
那就先快乐摸鱼去了~
吃完饭,OJ就好了,继续继续~
- 最直接的朴素做法:用k个空间存储数量为k的大米袋数,然后就转化为01背包
/**
* TODO Auther: by SR
* Date: 2020-01-23 22:40:19
* LastEditTime: 2020-01-24 13:52:50
* ! 今日运势: 吉,无bug
* ?多重背包问题
* *求最大值
* *这里是最朴素的求法
**/
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<string>
#include<memory.h>
using namespace std;
struct rice{
int p,h,c;
}a[2004];
void multiple_bag();
void split(int);
int ans[102];
int C,money,m,num;
int main(){
cin >> C;
while(C--){
ans[0] = 0, num = 0;
memset(ans,0,sizeof(ans));
cin >> money >> m;
for(int i = 1; i <= m; i++){
cin>>a[i].p>>a[i].h>>a[i].c;
split(i);
}
multiple_bag();
}
return 0;
}
void split(int now){
for(int i = 2; i <= a[now].c; i++){
a[m+(++num)] = a[now];
}
}
void multiple_bag(){
for(int i = 1; i <= m+num; i++){
for(int j = money; j >= a[i].p; j--){
ans[j] = max(ans[j], ans[j-a[i].p] + a[i].h);
}
}
int large = 0;
cout << ans[money] <<"\n";
return;
}
- 用二进制分割优化后:
/**
* TODO Auther: by SR
* Date: 2020-01-23 22:40:19
* LastEditTime: 2020-01-24 14:53:51
* ! 今日运势: 吉,无bug
* ?多重背包问题
* *求最大值
* TODO 用二进制拆分进行优化
**/
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<string>
#include<memory.h>
using namespace std;
struct rice{
int p,h,c;
}a[2004];
void multiple_bag();
void binary_split(int);
int ans[102];
int C,money,m,num;
int main(){
cin >> C;
while(C--){
ans[0] = 0, num = 0;
memset(ans,0,sizeof(ans));
cin >> money >> m;
for(int i = 1; i <= m; i++){
cin>>a[i].p>>a[i].h>>a[i].c;
binary_split(i);
}
/*
for(int i = 0; i <= m+num; i++){
printf("%d: %d %d\n",i,a[i].p,a[i].h);
}*/
multiple_bag();
}
return 0;
}
void binary_split(const int now){
int temp = 0,total = 0;
do{
a[m+(++num)].p = pow(2,temp)*a[now].p;
a[m+num].h = pow(2,temp)*a[now].h;
total += pow(2,temp);
temp++;
}while(a[now].c -total >= pow(2,temp));
if(a[now].c-total){
a[m+(++num)].p = (a[now].c - total)*a[now].p;
a[m+num].h = (a[now].c - total)*a[now].h;
}
}
void multiple_bag(){
for(int i = m+1; i <= m+num; i++){
for(int j = money; j >= a[i].p; j--){
ans[j] = max(ans[j], ans[j-a[i].p] + a[i].h);
}
}
int large = 0;
cout << ans[money] <<"\n";
return;
}
end