HDU 5550 dp + 前缀和优化

题意：

有个大楼有n层，每层都有两种人，一种喜欢的打球，一种喜欢游泳，现在要你在每一层要么开设球馆，要么开设游泳馆，分配完毕以后，打球的要到球馆，游泳的要到游泳馆，问最优分配下，所有人要移动的最短距离和是多少？

思路：

dp[i][0/1]表示第i层布置为0或1，且i+1层为1或0（与第i层相反）的最小代价。
dp[i][0]可以由dp[j][1]转移过来，其中j小于i，那么我们考虑j+1到i的这些层要建成0，考虑使得代价最小，那么这些层所有1的人都要分别向两边离开。比如3，4，5，6层，3层和4层的1会去2层，5层和6层的1要去7层。
这样状态转移方程为dp[i][0/1] = dp[j][1/0] + tmp。这里tmp的计算可以利用前缀和优化。使得整个算法复杂度为O（n^2）。
另外要注意边界处理，具体实现看代码：

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 4005;
const ll INF = 0x3f3f3f3f3f3f3f3f;

ll a[MAXN][2], dp[MAXN][2], sum[MAXN][2];

int main() {
    //freopen("in", "r", stdin);
    //freopen("out", "w", stdout);
    int T, cs = 0;
    scanf("%d", &T);
    while (T--) {
        int n;
        scanf("%d", &n);
        for (int i = 1; i <= n; i++) {
            scanf("%lld%lld", &a[i][0], &a[i][1]);
            sum[i][0] = sum[i - 1][0] + a[i][0];
            sum[i][1] = sum[i - 1][1] + a[i][1];
        }
        ll ans = INF;
        for (int i = 1; i < n; i++) {
            for (int k = 0; k < 2; k++) {
                dp[i][k] = 0;
                for (int j = 1; j <= i; j++)
                    dp[i][k] += a[j][k ^ 1] * (i - j + 1);
                ll tmp = a[i][k ^ 1];
                if (i - 1 >= 1) dp[i][k] = min(dp[i][k], dp[i - 1][k ^ 1] + tmp);
                for (int j = i - 2; j >= 1; j--) {
                    int mid = (i - j + 1) / 2 + j;
                    tmp += sum[mid][k ^ 1] - sum[j][k ^ 1];
                    dp[i][k] = min(dp[i][k], dp[j][k ^ 1] + tmp);
                }
                ll tt = 0;
                for (int j = i + 1; j <= n; j++) tt += a[j][k] * (j - i);
                //cout << tt << endl;
                ans = min(ans, dp[i][k] + tt);
            }
        }
        printf("Case #%d: %lld\n", ++cs, ans);
    }
    return 0;
}