化简以后就是裸的斜率优化dp了,如果不清楚斜率优化的话推荐去写玩具装箱,倒不是简单一些,只是题目老一点,有很多关于斜率优化的介绍
#include<cstdio>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
LL f[3020][3020], sum[3020];
int q[6020],n,m;
double Q(int a,int i,int j){
return ((double)(f[a][j-1]-f[i][j-1]+sum[a]*sum[a]-sum[i]*sum[i]))/(sum[a]-sum[i]);
}
int main(){
scanf("%d%d",&n,&m);
memset(f,63,sizeof(f));
f[0][0]=0;
for(int i=1;i<=n;i++)scanf("%lld",sum+i),sum[i]+=sum[i-1];
for(int j=1;j<=m;j++){
int l=1,r=1;q[l]=0;
for(int i=1;i<=n;i++){
while(l<r&&Q(q[l+1],q[l],j)<(double)2*sum[i])l++;
f[i][j]=f[q[l]][j-1]+(sum[i]-sum[q[l]])*(sum[i]-sum[q[l]]);
while(l<r&&Q(q[r-1],q[r],j)>=Q(q[r],i,j))r--;
q[++r]=i;
}
}
printf("%lld",f[n][m]*m-sum[n]*sum[n]);
return 0;
}