题意
东东开车出去泡妞(在梦中),车内提供了 n 张CD唱片,已知东东开车的时间是 n 分钟,他该如何去选择唱片去消磨这无聊的时间呢
假设:
CD数量不超过20张
没有一张CD唱片超过 N 分钟
每张唱片只能听一次
唱片的播放长度为整数
N 也是整数
我们需要找到最能消磨时间的唱片数量,并按使用顺序输出答案(必须是听完唱片,不能有唱片没听完却到了下车时间的情况发生)
Input
多组输入
每行输入第一个数字N, 代表总时间,第二个数字 M 代表有 M 张唱片,后面紧跟 M 个数字,代表每张唱片的时长 例如样例一: N=5, M=3, 第一张唱片为 1 分钟, 第二张唱片 3 分钟, 第三张 4 分钟
所有数据均满足以下条件:
N≤10000
M≤20
Output
输出所有唱片的时长和总时长,具体输出格式见样例
Example
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
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背包总结
- 设计状态: 表示仅考虑前 i 件物品,放入一个容量为 j 的背包可以获得的最大价值。
- 状态转移方程
- “前 i 件物品放入容量为 j 的背包中”这个子问题中,我们现在考虑第 i件物品的策略(放、不放)
- 如果选择不放,那么问题转化为“前 i-1件物品放入容量为 j 的背包中”,也就是 f[i-1][j] 所表示的子问题的状态;
- 如果选择放,那么问题转化成“前 i-1 件物品放入容量为 j-w[i] 的背包
中”,此时能获得的最大价值就是 f[i-1][j-w[i]] + v[i] - 两种策略,决策为取价值更大的策略
优化——滚动数组
- 如果只用一个数组 f[V],能不能保证第 i 次循环结束后 f[j] 中表示的就是我们定义的状态 f[i][j]?
- f[i][j] 是由 f[i-1][j] 和 f[i-1][j-w[i]] 两个子问题递推过来
- 只需要保证枚举顺序从 j = V → 0 即可
- 滚动数组的关键点是:逆序! 如果是顺序的话则会出现同一件物品选多次的情况
- 代码
思想
- 在此题中价值和质量是等价的,因此不妨吧价值和质量用一个数组存放
- 01背包问题的状态转移方程为: dp[i][j] = max { dp[i-1][j] , dp[i-1][j-w[i]]+c[i] }。在此基础上,开了一个bool类型的choice二维数组,用来记录计算 dp[i][j]时选择了那一个策略。如果状态转移时选择了 dp[i-1][j],那么记choice[i][j]=0 (即不放第i间物品);如果选择了 dp[i-1][j-w[i]]+c[i],记choice[i][j]=1 (即放第i间物品)。
- 这样当dp数组求解完毕后
- 先求出dp[m][0…j]中最大的dp[n][j]
- 然后按要求输出
代码
#include <iostream>
#include <string>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <stdio.h>
using namespace std;
int w[10010],dp[10010]={0};
bool choice[100][10010],flag[10010];
int main()
{
int m,n;
while (~scanf("%d%d",&n,&m))
{
memset(choice,0,sizeof(choice));
memset(flag,0,sizeof(flag));
memset(dp,0,sizeof(dp));
for(int i=1;i<=m;i++)
scanf("%d",&w[i]);
for(int i=1;i<=m;i++)
{
for(int j=n;j>=w[i];j--)
{//状态转移方程
if(dp[j]<dp[j-w[i]]+w[i])
{
dp[j]=dp[j-w[i]]+w[i];
choice[i][j]=1; //放入第i件物品
}
else choice[i][j]=0;
}
}
int k=m,j=n;
while(k>=0)
{
if(choice[k][j]==1)
{
flag[k]=true;
j-=w[k];
}
else flag[k]=false;
k--;
}
for(int i=1;i<=m;i++)
{
if(flag[i]==true)
{
printf("%d ",w[i]);
}
}
printf("sum:%d\n",dp[n]);
}
return 0;
}