题目链接: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
代码:
#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原创,但可以转载,因为我们是兄弟。