Uva-624 CD (需要输出方案的01背包)

#include <bits/stdc++.h>
using namespace std;
int dp[21][2002], a[102], pre[21][2002];
int main(){
    int n, m, x, ans;
    while(scanf("%d %d", &m, &n) != EOF){
        memset(dp, 0, sizeof(dp));
        memset(pre, 0, sizeof(pre));
        for(int i = 1; i <= n; ++i){
            scanf("%d", &a[i]);
        }
        for(int i = 1; i <= n; ++i){
            for(int j = m; j >= 0; --j){
                dp[i][j] = dp[i - 1][j];
                pre[i][j] = j;
                if(a[i] <= j && dp[i][j] < dp[i - 1][j - a[i]] + a[i]){
                    dp[i][j] = dp[i - 1][j - a[i]] + a[i];
                    pre[i][j] = j - a[i];
                }
            }
        }
        int pos = 0, ans = 0;
        for(int i = 0; i <= m; ++i){
            if(ans < dp[n][i]){
                ans = dp[n][i];
                pos = i;
            }
        }
        for(int i = n; i >= 1; --i){
            if(pre[i][pos] != pos){
                printf("%d ", a[i]);
                pos -= a[i];
            }
        }
        printf("sum:%d\n", ans);
    }
}

/*
题意：20个数中选一些数出来，使得和最大且不超过m。输出方案。

思路：这题就是裸的背包加一个方案输出。设dp[i][j]表示前i个数字和上限为j时的最优方案和。pre[i][j]表示前i个
最优方案和为j是从多少转移来的，如果pre[i][j] == j，说明我没有选第i个，如果pre[i][j] == j - a[i]说明选了。
转移一下就好了。
*/