HDU 4261 Estimation（set预处理中位数+dp+缺页中断t个爽）



题意：给出一个序列A，分为K个部分，然后每个部分给出一个B（自己给出），使得所有的∑(|Ai-Bi|)最小。

思路：首先这道题看起来比较像 POJ 1260

也就是说任意一堆数中取一个使得∑(|Ai-B|)最小，那么我们肯定取这些数的中位数作为B，

但是这道题与poj 1260不同的是这里的数是不连续的，也就是说我们要动态维护一个中位数，我们可以用两个set s1,s2来维护，其中s2中的元素总比s1中的大且s2的容量总比s1大一或相等，再维护set元素的同时我们同时维护这两个set中的元素和，这个set维护的思路类似于百度之星预赛的一道题 2015百度之星初赛1 1004 kpi题解（set的运用）。

那么剩下来的就是DP了，一开始dp的二维数组状态开反了，导致内存访问不连续缺页中断t个爽......

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<string>
#include<map>
#include<set>
#include<ctime>
#define eps 1e-6
#define LL long long
#define pii pair<int, int>
//#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;

const int MAXN = 2100;
const int INF = 0x3f3f3f3f;
int n, K, sum1, sum2;
int a[MAXN], d[MAXN][MAXN], dp[30][MAXN];
multiset<int> s1, s2;
void init_d() {
	for(int i = 1; i <= n; i++) {
		s1.clear(); s2.clear();
		sum1 = sum2 = 0;
		for(int j = i; j <= n; j++) {
			int sz1 = s1.size(), sz2 = s2.size();
			if(sz1 == sz2) {
				if(!sz1) {
					s2.insert(a[j]);
					sum2 += a[j];
					continue;
				}
				int tmp = *s1.rbegin();
				if(tmp <= a[j]) s2.insert(a[j]), sum2 += a[j];
				else {
					s1.erase(s1.find(tmp));
					s1.insert(a[j]);
					sum1 += a[j] - tmp;
					s2.insert(tmp);
					sum2 += tmp;
				}
			}
			else {
				int tmp = *s2.begin();
				if(a[j] > tmp) {
					s1.insert(tmp);
					sum1 += tmp;
					s2.erase(s2.begin());
					s2.insert(a[j]);
					sum2 += a[j] - tmp;
				}
				else s1.insert(a[j]), sum1 += a[j];
			}
			d[j][i] = sum2-(j-i+2)/2*(*s2.begin()) + (j-i+1)/2*(*s2.begin())-sum1;
		}
	}
}
void DP() {
	for(int i = 1; i <= n; i++) dp[1][i] = d[i][1];
	for(int k = 2; k <= K; k++) {
		for(int i = k; i <= n; i++) {
			dp[k][i] = INF;
			for(int j = k-1; j < i; j++) {
				dp[k][i] = min(dp[k][i], dp[k-1][j]+d[i][j+1]);
			}
		}
	}
}
int main() {
    //freopen("input.txt", "r", stdin);
	while(scanf("%d%d", &n, &K)==2 && n) {
		for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
		init_d();
		DP();
		printf("%d\n", dp[K][n]);
	}
    return 0;
}