题意:
东东开车出去泡妞(在梦中),车内提供了 n 张CD唱片,已知东东开车的时间是 n 分钟,他该如何去选择唱片去消磨这无聊的时间呢
假设:
·CD数量不超过20张
·没有一张CD唱片超过 N 分钟
·每张唱片只能听一次
·唱片的播放长度为整数
·N 也是整数
我们需要找到最能消磨时间的唱片数量,并按使用顺序输出答案(必须是听完唱片,不能有唱片没听完却到了下车时间的情况发生)
多组输入
每行输入第一个数字N, 代表总时间,第二个数字 M 代表有 M 张唱片,后面紧跟 M 个数字,代表每张唱片的时长 例如样例一: N=5, M=3, 第一张唱片为 1 分钟, 第二张唱片 3 分钟, 第三张 4 分钟
所有数据均满足以下条件:
N≤10000
M≤20
输出所有唱片的时长和总时长,格式为 时长 时长 sum:总时长。
思路:
这是一道0-1背包问题,n个物品对应M张唱片,背包体积V对应开车时间N,物品的体积w[i]和价值v[i]都是唱片时长。要求最终背包容量至多为V,且要按顺序输出所放物品。
于是在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背包问题,且要记录所放物品的顺序。只需要在0-1背包的基础上稍加修改即可。
代码:
#include <iostream>
#include <vector>
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(cin>>V)
{
cin>>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;
}
}