F-选做题11-2(东东开车了)
一、题目描述
东东开车出去泡妞(在梦中),车内提供了 n 张CD唱片。
已知东东开车的时间是 n 分钟,他该如何去选择唱片去消磨这无聊的时间呢?
假设:
CD数量不超过20张
没有一张CD唱片超过 N 分钟
每张唱片只能听一次
唱片的播放长度为整数
N 也是整数
我们需要找到最能消磨时间的唱片数量,并按使用顺序输出答案
(必须是听完唱片,不能有唱片没听完却到了下车时间的情况发生)
本题是 Special Judge
Input
多组输入
每行输入第一个数字N, 代表总时间,第二个数字 M 代表有 M 张唱片,
后面紧跟 M 个数字,代表每张唱片的时长
例如样例一: N=5, M=3, 第一张唱片为 1 分钟, 第二张唱片 3 分钟, 第三张 4 分钟
所有数据均满足以下条件:
N≤10000
M≤20
Output
输出所有唱片的时长和总时长,具体输出格式见样例
Sample Input
5 3 1 3 4
10 4 9 8 4 2
20 4 10 5 7 4
90 8 10 23 1 2 3 4 5 7
45 8 4 10 44 43 12 9 8 2
Sample Output
1 4 sum:5
8 2 sum:10
10 5 4 sum:19
10 23 1 2 3 4 5 7 sum:55
4 10 12 9 8 2 sum:45
二、思路与算法
本题主要是01背包,但是要输出路径。
首先利用滚动数组计算01背包,在计算的同时记录该元素是否被加入路径中。
即如果f[j]<f[j-v[i]]+v[i]时,那么证明当前这个元素要被选定,path[i][j]=1。
确定最终路径时,一定要从后向前找路径中的元素!因为有的元素当时被选定,但最终可能不被选定,所以从后向前推,得到的元素都是得到最终结果所需要的。
我的代码中用vector类型的p把倒推出的路径逆序输出,与给出的样例完全一致。
但是因为是special judge,和样例顺序不同也可以,直接在倒推的过程中输出路径也可以AC。
同时,最后的sum相同,路径也可能会不同,不需要因为路径不同而纠结。
写完之后才发现,既然要申请path数组,不如直接就不用滚动数组,也就是用一个这么大的数组。
所以用滚动数组写反而又浪费时间空间,还麻烦了。
三、代码实现
#include<cstdio>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;
int N=0,M=0;
int ans=0;
int spr[22]={0};
int f[10010]={0};
int path[22][10010]={0};
vector<int> p(0);
int main(){
while(scanf("%d %d",&N,&M)!=EOF){
for(int i=1;i<=M;i++){
scanf("%d",&spr[i]);
}
memset(f,0,sizeof(f));
memset(path,0,sizeof(path));
p.clear();
//0-1背包
for(int i=1;i<=M;i++){
for(int j=N;j>=spr[i];j--){
if(j-spr[i]>=0){
if(f[j]<f[j-spr[i]]+spr[i]){
f[j]=f[j-spr[i]]+spr[i];
path[i][j]=1;
}
}
}
}
ans=f[N];
for(int i=M,j=N;i>=1&&j>0;i--){
if(path[i][j]){
p.push_back(spr[i]);
j-=spr[i];
}
}
//逆序输出
for(int i=p.size()-1;i>=0;i--){
printf("%d ",p[i]);
}
printf("sum:%d\n",ans);
}
return 0;
}