题目链接:http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=565
题目大意:很标准的01背包,但要输出路径。
解题思路:由于要输出路径,不能再用滚动数组,要用二维数组,至少我没想到怎么用滚动数组实现。在转移的时候记录下path[i][j],如果选择,那么它为1,否则为0.那么在递归输出路径的时候,就可以根据这个物品是否使用,然后j减去cost[i]就是上一个容量。具体见代码,不影响O(V*N)的复杂度。
测试数据:
5 3 1 3 4
10 4 9 8 4 2
45 8 4 10 44 43 12 9 8 2
1 1 1
1 1 2
代码:
#include <stdio.h>
#include <string.h>
#define MAX 110000
int n,m,time[100],ans[100];
int maxx,path[30][MAX],dp[30][MAX];
int main()
{
int i,j,k;
while (scanf("%d",&m) != EOF) {
scanf("%d",&n);
for (i = 1; i <= n; ++i)
scanf("%d",&time[i]);
memset(dp,0,sizeof(dp));
memset(path,0,sizeof(path));
dp[0][0] = 1;
for (i = 1; i <= n; ++i) {
for (j = m; j >= time[i]; --j)
if (dp[i-1][j-time[i]]) dp[i][j] = 1;
for (j = 0; j <= m; ++j)
if (dp[i][j] == 1) path[i][j] = 1;
else if (dp[i-1][j]) dp[i][j] = 1;
}
for (j = m; j >= 1; --j)
if (dp[n][j]) break;
maxx = j,i = n,k = 0;
while (i > 0) {
if (path[i][j] != 0) {
ans[++k] = i;
j = j - time[i];
}
i = i - 1;
}
for (i = k; i >= 1; --i)
printf("%d ",time[ans[i]]);
printf("sum:%d\n",maxx);
}
}
本文ZeroClock原创,但可以转载,因为我们是兄弟。