HDU4261 Estimation(dp,对顶堆)

传送门
一道dp水题.这32s的时限真是活久见.考虑用dp[i][j]表示前i个数分了j组能得到的minvalue.那么转移方程就可以列出来:
dp[i][j] = min(dp[k][j-1] + value(k+1,j))
意思是把k+1-i划分为新的一组.那么关键就在计算这个value函数.
很显然,我们只要找到k+1-i的中位数作为b[k]就是最小的代价.这个过程可以用一个对顶堆预处理出来,找到之后还要计算代价.如果暴力的去算的话预处理的复杂度是O(n³)的,虽然有32s的时限也依然会超时.所以需要优化一下.其实也很简单,利用对顶堆的性质,可以把绝对值去掉直接O(1)的计算代价.具体实现看代码吧.

#pragma GCC optimize(2)
#define LL long long
#define pq priority_queue
#define ULL unsigned long long
#define pb push_back
#define mem(a,x) memset(a,x,sizeof a)
#define pii pair<int,int>
#define fir(i,a,b) for(int i=a;i<=(int)b;++i)
#define afir(i,a,b) for(int i=(int)a;i>=b;--i)
#define ft first
#define vi vector<int>
#define sd second
#define ALL(a) a.begin(),a.end()
#define bug puts("-------")
#define mpr(a,b) make_pair(a,b)
#include <bits/stdc++.h>

using namespace std;
const int N = 2e3+10;

inline void read(int &a){
	int x = 0,f=1;char ch = getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
	a = x*f;
}

int n,a[N],m,dp[N][27],val[N][N];

void prework(){
	fir(i,1,n){
		pq<int> q1;
		pq<int,vector<int>,greater<int> > q2;
		int sum1 = 0,sum2 = 0;
		fir(j,i,n){
			if(q1.empty() || a[j] <= q1.top()) q1.push(a[j]),sum1+=a[j];
			else q2.push(a[j]),sum2+=a[j];
			while(q1.size() && (int)q1.size() - (int)q2.size() > 1){
				q2.push(q1.top());
				sum2+=q1.top();sum1-=q1.top();
				q1.pop();
			}
			while(q2.size() && (int)q2.size() - (int)q1.size() > 0){
				q1.push(q2.top());
				sum1+=q2.top();sum2-=q2.top();
				q2.pop();
			}
			int v = q1.top();
			int s1 = q1.size(),s2 = q2.size();
			val[i][j] = (s1-s2)*v - sum1 + sum2 ;
		}
	}
}
int main(){
	while(cin >> n >> m, n && m){
		mem(dp,0x3f);
		dp[0][0] = 0;
		fir(i,1,n) cin >> a[i];
		prework();
		fir(i,1,n){
			fir(j,1,min(i,m)){
				fir(k,0,i-1){
					dp[i][j] = min(dp[i][j],dp[k][j-1]+val[k+1][i]);
				}
			}
		}
		cout << dp[n][m] << endl;
	}	
	
	return 0;
}	

/*
	dp[i][j] 表示把前 i 个数分为 j 组的minvalue.
*/