WEEK 11 F 选做题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

思路

这其实是一个0-1背包问题，背包容量为总时长，每个物品的体积为CD的时长，唯一不同的是需要保存目前已选择的每张唱片的时长。
回溯过程的实现就是考虑当前dp[x][y]，如果dp[x][y]=dp[x-1][y]，那就说明没有选择第x个物品，回溯x-1, y。如果dp[x][y]=dp[x-1][y-w[i]]+v[i]，那么就说明选择了第x件物品，回溯x-1, y-w[i]。直到x和y有一方为0就返回。

代码

#include <iostream>
#include <string>
#include <string.h>
#include <algorithm>

using namespace std;

const int N = 21;
const int M = 10001;
int n, m;
int f[N], dp[N][M];

void back(int x, int y) 
{
	if (x == 0 || y == 0) 
		return;
	if (dp[x][y] == dp[x - 1][y])
		back(x - 1, y);
	else if (dp[x][y] == dp[x - 1][y - f[x]] + f[x]) 
	{
		back(x - 1, y - f[x]);
		printf("%d ", f[x]);
	}
}

int main() 
{
	while (scanf("%d%d",&n,&m)!=EOF)
	{
		memset(dp, 0, sizeof(dp));
		for (int i = 1; i <= m; i++)
			scanf("%d", &f[i]);
		for (int i = 1; i <= m; i++)
			for (int j = 0; j <= n; j++) 
			{
				if (j >= f[i]) 
					dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - f[i]] + f[i]);
				else 
					dp[i][j] = dp[i - 1][j];
			}
		back(m, n);
		printf("sum:%d\n", dp[m][n]);
	}
	return 0;
}