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]);
}