HDU 5550 (DP)

题目链接：点击打开链接

题意：有n层楼，每层有两类人，假设是0，1，然后为每一层楼设置一个01属性使得

所有人走到自己属性的楼层的距离和最小。

容易想到的是dp[i][0/1]表示这一层属性是0/1的情况下1-i层的最小值，但是有个问题就是

另一种属性不知道应该去那一层，所以状态转变一下。因为层的属性肯定是连续的0或者

1拼接成的，可以设dp[i][0/1]表示这一层属性是0/1的情况并且和前一层状态不同，1-i层的

最小值。

对于某一楼i假如是0，那么它之前肯定有一段1，然后就枚举这个1的状态转变楼层j，那么

j-1必然是0，然后转移的时候，i~j之间的1属性的人只需要i层里面的往下走一楼，0属性的

人可以分成两半，前一半走到i，后一半走到j-1。另一种情况也类似。

然后需要注意的是一个区间的人全部走到某一层。容易搞的是开一个二维数组，然

而MLE。那就只有本方法了，各种前缀和弄起来~

吐槽一下，自己写的代码真是丑哭，自带大常数大模拟光环~~~~

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <iostream>
using namespace std;
#define maxn 4111
#define INF 111111111111111111

long long dp[maxn][2];
long long sum1[maxn][2], sum2[maxn][2];//0表示到n 1表示到1
long long all1[maxn], all2[maxn];//前缀和
long long a[maxn], b[maxn];
int n;

long long get_up (int i, int j, bool op) {//向上都跑到i
	if (i == j) return 0;
	long long ans = 0;
	if (!op) {
		long long tmp = (j == 1 ? all1[i-1] : all1[i-1]-all1[j-1]);
		long long cur = sum1[j][0]-sum1[i][0];
		ans = cur-tmp*(n-i);
	}
	else {
		long long tmp = (j == 1 ? all2[i-1] : all2[i-1]-all2[j-1]);
		long long cur = sum2[j][0]-sum2[i][0];
		ans = cur-tmp*(n-i);
	}
	return ans;
}

long long get_down (int i, int j, int op) {//向下都跑到i
	if (i == j) return 0;
	long long ans = 0;
	if (!op) {
		long long tmp = all1[j]-all1[i];
		long long cur = sum1[j][1]-sum1[i][1];
		ans = cur-tmp*(i-1);
	}
	else {
		long long tmp = all2[j]-all2[i];
		long long cur = sum2[j][1]-sum2[i][1];
		ans = cur-tmp*(i-1);
	}
	return ans;
}

int main () {
    //freopen ("in.txt", "r", stdin);
    int t, kase = 0;
    scanf ("%d", &t);
    while (t--) {
        scanf ("%d", &n);
        all1[0] = all2[0] = 0;
        for (int i = 1; i <= n; i++) {
            scanf ("%lld%lld", &a[i], &b[i]);
            all1[i] = all1[i-1]+a[i];
            all2[i] = all2[i-1]+b[i];
        }
        for (int i = n; i >= 1; i--) {
        	if (i == n) 
        		sum1[i][0] = sum2[i][0] = 0;
        	else {
        		sum1[i][0] = sum1[i+1][0]+a[i]*(n-i);
        		sum2[i][0] = sum2[i+1][0]+b[i]*(n-i);
        	}
        }
        for (int i = 1; i <= n; i++) {
        	if (i == 1) 
        		sum1[i][1] = sum2[i][1] = 0;
        	else {
        		sum1[i][1] = sum1[i-1][1]+a[i]*(i-1);
        		sum2[i][1] = sum2[i-1][1]+b[i]*(i-1);
        	}
        }
        for (int i = 2; i <= n; i++) {
            dp[i][0] = get_up (i, 1, 0)+b[i];
            dp[i][1] = get_up (i, 1, 1)+a[i];
        }
        for (int i = 3; i <= n; i++) {
            for (int j = i-1; j > 1; j--) {
                double mid = (i+j-1)/2.0;
                int up = ceil (mid), down = floor (mid);
                if (up == down)
                    up++;
                dp[i][0] = min (dp[i][0], dp[j][1]+b[i]+get_up (i, up, 0)+get_down (j-1, down, 0)-a[j]);
                dp[i][1] = min (dp[i][1], dp[j][0]+a[i]+get_up (i, up, 1)+get_down (j-1, down, 1)-b[j]);
            }
        }

        long long ans = INF;
        for (int i = 2; i <= n; i++) {
            ans = min (ans, dp[i][0]+get_down (i-1, n, 1)-b[i]);
            ans = min (ans, dp[i][1]+get_down (i-1, n, 0)-a[i]);
        }
        printf ("Case #%d: %lld\n", ++kase, ans);
    }
    return 0;
}
/*
3
4
1 1
1 1
1 1
1 1
4
1 2
1 2
1 2
1 2
2
10 5
4 3
*/