BZOJ2933: [Poi1999]地图

BZOJ2933: [Poi1999]地图

https://lydsy.com/JudgeOnline/problem.php?id=2933

分析:

• 题中的那个\(A(k)\)显然就是中位数。
• 排序后肯定是一段一段选，然后就做完了。

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cmath>
using namespace std;
#define N 3050
typedef long long ll;
int n,m,a[N];
ll f[11][N],s[N],c[N][N];
int main() {
	scanf("%d%d",&n,&m);
	int i,j,k;
	for(i=1;i<=n;i++) scanf("%d",&a[i]);
	sort(a+1,a+n+1);
	memset(f,0x3f,sizeof(f));
	f[0][0]=0;
	for(i=1;i<=n;i++) s[i]=s[i-1]+a[i];
	for(i=1;i<=n;i++) {
		c[i][i]=0;
		for(j=i+1;j<=n;j++) {
			int p=(i+j)>>1;
			c[i][j]=ll(a[p])*(p-i+1)-(s[p]-s[i-1])+(s[j]-s[p-1])-ll(a[p])*(j-p+1);
		}
	}
	for(i=1;i<=m;i++) {
		for(j=1;j<=n;j++) {
			ll tmp=1ll<<60;
			for(k=0;k<j;k++) {
				tmp=min(tmp,f[i-1][k]+c[k+1][j]);
			}
			f[i][j]=tmp;
		}
	}
	printf("%lld\n",f[m][n]);
}