【线性递推DP】hdu 5550

2015南阳CCPC的银牌题。

题意

这里曲折一下原本故事背景。
有n层楼，每层楼只有一个男厕所或者女厕所，现在每层楼的男生有ti人，女生有pi人。
他们每往上走一层或者往下走一层的花费是1个单位。
现在问你应该怎么建厕所才能使总体的花费最少。

思路

不会写啊，怎么推都是一个$O(n^3)$的DP，优化不下来。
于是参考了
牛犇的代码

思路是这样子的。
dp[第几层楼][所建的性别]
表示，前i层楼，在第i层楼建sex厕所的情况下。最少的代价是多少
我们特别的考虑到，第i层以后的，他们都建跟第i层不一样的厕所。
那么可以得到$ans = min(ans, dp[i][sex] + godown(i, n, sex));$

这里的godown和goup函数，是特别计算。
goup：性别为sex的人，全都走到l的花费是多少
godown：性别为sex的人全都走到r的花费是多少
那么cnt就是求。l到r区间，这些人去到他们厕所的最短路程。

接下来我们每次只要假设。
第j+1层和第i层中间都是同性别的厕所。将他们不同性别的人都走到j或i层即可。

#include <bits/stdc++.h>

#define ll long long
#define MAXN 4005
#define INF 1e18

using namespace std;

struct node {
    ll t, p;
} pot[MAXN];

ll sum[MAXN][2], dsum[MAXN][2];

ll dp[MAXN][2];

void init() {
    for (int i = 0; i < MAXN; i++) {
        dp[i][0] = dp[i][1] = INF;
    }
}

void output(ll *num, int len) {
    for (int i = 1; i <= len; i++) {
        printf("%lld ", num[i]);
    }
    printf("\n");
}

ll goup(int l, int r, int sex) {
    return (sum[r][sex] - sum[l][sex]) * (r + 1) - (dsum[r][sex] - dsum[l][sex]);
}

ll godown(int l, int r, int sex) {
    return dsum[r][sex] - dsum[l][sex] - (sum[r][sex] - sum[l][sex]) * l;
}

ll cnt(int l, int r, int sex) {
    int mid = l + r >> 1;
    return godown(l, mid, sex) + goup(mid, r, sex);
}

int main() {
    int T, n, cas = 1;
    scanf("%d", &T);
    while (T--) {
        init();
        scanf("%d", &n);
        for (int i = 1; i <= n; i++) {
            scanf("%lld %lld", &pot[i].t, &pot[i].p);
        }
        sum[0][0] = sum[0][1] = dsum[0][0] = dsum[0][1] = 0;
        for (int i = 1; i <= n; i++) {
            sum[i][0] = sum[i - 1][0] + pot[i].t;
            sum[i][1] = sum[i - 1][1] + pot[i].p;

            dsum[i][0] = dsum[i - 1][0] + pot[i].t * i;
            dsum[i][1] = dsum[i - 1][1] + pot[i].p * i;
        }
        ll ans = INF;
        for (int i = 1; i < n; i++) {
            dp[i][0] = goup(0, i, 1);
            dp[i][1] = goup(0, i, 0);
            for (int j = 1; j < i; j++) {
                dp[i][0] = min(dp[i][0], dp[j][1] + cnt(j, i, 1));
                dp[i][1] = min(dp[i][1], dp[j][0] + cnt(j, i, 0));
            }
            ans = min(ans, dp[i][0] + godown(i, n, 0));
            ans = min(ans, dp[i][1] + godown(i, n, 1));
        }
        printf("Case #%d: %lld\n", cas++, ans);
    }
}

/*
100
4
1 2
2 1
1 2
1 2

3
10 5
4 3
10 5
*/