传送门
题解:先按照a排序,预处理前缀和,然后预处理区间代价(每次贪心地选中位数然后计算),最后dp一次即可。dp[i][j]表示前i个地区分成j个颜色的最小代价。
吐槽一下,有位大哥做bzoj3065一道时限60秒的题TLE了8次算不算卡评测啊(╯‵□′)╯︵┻━┻
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,m;
ll dp[3002][12],w[3002][3002],a[3002],sum[3002]={0};
int main() {
// freopen("bzoj 2933.in","r",stdin);
scanf("%d%d",&n,&m);
for (int i=1;i<=n;++i) scanf("%lld",&a[i]);
sort(a+1,a+n+1);
for (int i=1;i<=n;++i) sum[i]=sum[i-1]+a[i];
for (int i=1;i<=n;++i)
for (int j=i;j<=n;++j) {
int mid=(i+j)>>1;
w[i][j]=a[mid]*(mid-i)-(sum[mid-1]-sum[i-1])-a[mid]*(j-mid)+(sum[j]-sum[mid]);
}
memset(dp,127,sizeof(dp));
dp[0][0]=0;
for (int i=1;i<=n;++i)
for (int j=1;j<=m;++j)
for (int k=0;k<i;++k)
dp[i][j]=min(dp[i][j],dp[k][j-1]+w[k+1][i]);
printf("%lld\n",dp[n][m]);
return 0;
}