week11 F - 选做题11-2 东东开车了

选做题11-2 东东开车了

东东开车出去泡妞(在梦中)，车内提供了 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

我的思路：
这道题是一道需要输出路径的01背包问题。这里我在普通01背包问题解法的基础上加了一个记录路径的二维数组p。p记录了相关层要更新的dp，即判断在哪里进行了哪个新唱片的加入。然后在输出路径的时候，只要找到p数组中变化的位置，输出对应的唱片即可。

我的总结：
这道题涉及到了路径的输出，在最普通的01背包问题的基础上，有了一定的提升。

我的代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

int M,N,Ch[21],dp[100010],p[100][100010];

int main()
{
	while(~scanf("%d%d",&N,&M))
	{
		for(int i=1;i<=M;i++) scanf("%d",&Ch[i]);
		memset(dp,0,sizeof(dp)); 
		memset(p,0,sizeof(p));
		for(int i=M;i>=1;i--)
		for(int j=N;j>=Ch[i];j--)
		{
			if(dp[j] < dp[j-Ch[i]]+Ch[i])
			{
				dp[j] = dp[j-Ch[i]]+Ch[i];
				 p[i][j] = 1;
			} 
		}
		for(int i=1,j=N;i<=M && j>0;i++)
		{
			if(p[i][j])
			{
				printf("%d ",Ch[i]);
				j -= Ch[i];
			}
		}
		printf("sum:%d\n",dp[N]);
	}
	
	return 0;
}