题目描述:
东东开车出去泡妞(在梦中),车内提供了 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:
输出所有唱片的时长和总时长,具体输出格式见样例
思路:
01背包问题,需要输出路径。
记录前驱bool pre[i][j]==1 为当前f[i][j]装入了第i件物品。i从n件物品、j最大空间m开始循环,若pre[i][j]==1,则j更新为j-weight[i]。每次循环后i--,表示向前遍历物品。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int a[10010],v[10010],f[10010];
bool pre[10010][10010];
int main()
{
int m,n;
while(scanf("%d",&m)!=EOF)
{
memset(f,0,sizeof(f));
memset(pre,0,sizeof(pre));
cin>>n;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<=n;i++)
for(int j=m;j>=0;j--)
if(j-a[i]>=0)
if(f[j]<=f[j-a[i]]+a[i])
{
f[j]=f[j-a[i]]+a[i];
pre[i][j]=1;
}
int i=n,j=m,sum=0;
while(i>=1&&j>=0)
{
if(pre[i][j])
{
j=j-a[i];
v[++sum]=a[i];
}
i--;
}
for(int i=sum;i>=1;i--)
cout<<v[i]<<" ";
cout<<"sum:"<<f[m]<<endl;
}
return 0;
}