题目描述
东东开车出去泡妞(在梦中),车内提供了 n 张CD唱片,已知东东开车的时间是 n 分钟,他该如何去选择唱片去消磨这无聊的时间呢
假设:
CD数量不超过20张
没有一张CD唱片超过 N 分钟
每张唱片只能听一次
唱片的播放长度为整数
N 也是整数
需要找到最能消磨时间的唱片数量,并按使用顺序输出答案(必须是听完唱片,不能有唱片没听完却到了下车时间的情况发生)
输入格式
多组输入
每行输入第一个数字N, 代表总时间,第二个数字 M 代表有 M 张唱片,后面紧跟 M 个数字,代表每张唱片的时长 例如样例一: N=5, M=3, 第一张唱片为 1 分钟, 第二张唱片 3 分钟, 第三张 4 分钟
所有数据均满足以下条件:
N≤10000
M≤20
输出格式
输出所有唱片的时长和总时长,具体输出格式见样例
输入样例
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
输出样例
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背包问题,背包容量为总时长,每个物品的体积为CD的时长。
唯一不同的是需要保存目前已选择的每张唱片的时长。
因此这里既可通过保存选择每张唱片时的状态数组倒推选择的唱片,也可给每个状态一个vector容器,来保存选择的唱片。这里使用了容器保存选择的唱片,同时使用数组复用来降低空间复杂度。
注意
这里是尽可能选择前面的唱片,如输入的最后一组,既可选择对应的输出组,亦可选择43 2 这两张唱片,但为了尽可能选择前面的唱片,所以选择了样例输出中的那组,因此设计状态转移时,当选择某一唱片时,选择之前的某一状态加上当前唱片的总时长与不选择当前唱片总时长相同时,需要尽可能选择当前唱片,详见代码(solve函数中的if判断语句中的<=号,即=时也需更新)。
代码
#include <iostream>
#include <vector>
using namespace std;
const int N=1e4+10;
const int M=20+10;
int t,n;
int CD[M];
pair<int,vector<int>> dp[N];
void solve()
{
for(int i=0;i<=t;i++)
dp[i].first=0,dp[i].second.clear();
for(int i=1;i<=n;i++)
{
for(int j=t;j>0;j--)
{
if((j-CD[i])>=0&&dp[j].first<=(dp[j-CD[i]].first+CD[i]))
{
dp[j].first=dp[j-CD[i]].first+CD[i];
dp[j].second=dp[j-CD[i]].second;
dp[j].second.push_back(CD[i]);
}
}
}
}
int main(int argc, char** argv) {
while(scanf("%d%d",&t,&n)!=EOF)
{
for(int i=1;i<=n;i++)
scanf("%d",&CD[i]);
solve();
for(auto it=dp[t].second.begin();it!=dp[t].second.end();it++)
printf("%d ",(*it));
printf("sum:%d\n",dp[t].first);
}
return 0;
}