LightOJ - 1126 Building Twin Towers(DP)

题目大意：给你N个积木，要求你使用这N个积木，垒成两个高度相同的塔，问最大的高度是多少

解题思路：用dp[i][j]表示使用前i个积木，垒成两座塔，高度差为j，最高的塔的高度
接着就可以转移了，一个是不使用积木，则dp[i][j] = dp[i-1][j]
一个是给最高的塔 dp[i][j + height[i]] = max(dp[i][j + height[i]], dp[i - 1][j] + height[i])
一个是给较低的塔，这得分两种情况，一种是使用该积木后，高度超过了最高的塔，另一种则是，还是低于最高的塔

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 52;
const int M = 500010;

int val[N];
int sum, cas = 1, n;
int dp[2][M];

void init() {
    sum = 0;
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &val[i]);
        sum += val[i];
    }
    sum = (sum + 1) / 2;
}

void solve() {
    memset(dp, -1, sizeof(dp));
    int flag = 0;
    dp[0][0] = 0;
    for (int i = 1; i <= n; i++) {
        for (int j = 0; j <= sum; j++) {
            dp[flag ^ 1][j] = max(dp[flag][j], dp[flag ^ 1][j]);
            if (~dp[flag][j]) {
                int t1 = dp[flag][j];
                int t2 = dp[flag][j] - j;

                //给最高的
                if (j + val[i] <= sum) dp[flag ^ 1][j + val[i]] = max(dp[flag ^ 1][j + val[i]], t1 + val[i]) ;
                //加上去还是低于最高的塔
                if (j >= val[i]) dp[flag ^ 1][j - val[i]] = max(dp[flag ^ 1][j - val[i]], t1);
                else if(val[i] - j <= sum) dp[flag ^ 1][val[i] - j] = max(dp[flag ^ 1][val[i] - j], t2 + val[i]);
            }
        }
        flag = flag ^ 1;
    }
    if (dp[flag][0] == 0) printf("Case %d: impossible\n", cas++);
    else printf("Case %d: %d\n", cas++, dp[flag][0]);
}

int main() {
    int test;
    scanf("%d", &test);
    while (test--) {
        init();
        solve();
    }
    return 0;
}