东东开车出去泡妞(在梦中),车内提供了 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
解题思路:
在0-1背包的滚动数组f[j]=max(f[j],f[j-w[i]]+v[i])的基础上,使用了vector ff[j]来记录所放物品的顺序。当要放入当前物品(即f[j-w[i]]+v[i]>f[j])时,要记录当前物品,即ff[j]=ff[j-w[i]],ff[j].push_back(v[i])。在0-1背包代码基础上添加存储物品的顺序即可。
#include <iostream>
#include <vector>
#include<cstdio>
using namespace std;
int n,V;
int v[21],w[21];
int f[10010];
vector<int> ff[10010];
void solve()
{
for(int i=0;i<=V;i++)
f[i]=0,ff[i].clear();
for(int i=1;i<=n;i++)
{
for(int j=V;j>=0;j--)
{
if(j-w[i]>=0)
{
if(f[j]<f[j-w[i]]+v[i])
{
f[j]=f[j-w[i]]+v[i];
ff[j]=ff[j-w[i]];
ff[j].push_back(v[i]);
}
}
}
}
}
int main()
{
while(scanf("%d",&V)!=EOF)
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
cin>>v[i],w[i]=v[i];
solve();
for(int i=0;i<ff[V].size();i++)
cout<<ff[V][i]<<" ";
cout<<"sum:"<<f[V]<<endl;
}
}