裸的斜率优化。。我考场上SB写暴力
#include<algorithm>
#include<iostream>
#include<complex>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<vector>
#include<cmath>
#include<queue>
#include<ctime>
#include<set>
#include<map>
#define N 3333
#define inf 1e9
using namespace std;
long long sum[N],a[N],b[N],*f=a,*g=b;
int q[N],n,m;
double slop(int x,int y)
{
return double(g[x]-g[y]+sum[x]*sum[x]-sum[y]*sum[y])/(sum[x]-sum[y]);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&sum[i]),sum[i]+=sum[i-1];
for(int i=1;i<=n;i++) f[i]=sum[i]*sum[i];
for(int o=1;o<m;o++)
{
int l=1,r=1; swap(f,g);q[1]=0;
for(int i=1;i<=n;i++)
{
while(l<r&&slop(q[l],q[l+1])<2*sum[i])l++;
f[i]=g[q[l]]+(sum[i]-sum[q[l]])*(sum[i]-sum[q[l]]);
while(l<r&&slop(q[r],q[r-1])>slop(q[r],i))r--;
q[++r]=i;
}
}
cout<<m*f[n]-sum[n]*sum[n];
}