式子化简后,就是求
m∗∑mi=1xi2−sum[n]2
的最小值。
容易有
f[i][j]=min(f[i−1][k]+(sum[j]−sum[k])2)
然后很容易得出具有决策单调性,可以斜率优化。
。。为什么我连队列和
l,r
都要开二维。。并不知道一维怎么做的。。
反正题目不卡空间,递推就懒得滚动数组了,还跑得慢。
【代码】
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <stack>
#include <cstring>
#include <vector>
#include <queue>
#define N 3005
#define M 200005
#define INF 0x7fffffff
using namespace std;
typedef long long ll;
typedef pair<int,int> pa;
ll read()
{
ll x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
return x*f;
}
int n,m,now;
int q[2][N],l[2],r[2];
ll sum[N],f[N][N];
ll Sqr(ll x){
return x*x;
}
double Slope(int i,int j,int k){
return (double)(f[i][k]+Sqr(sum[k])-f[i][j]-Sqr(sum[j]))/(double)(sum[k]-sum[j]);
}
int main()
{
n=read(),m=read();
for(int i=1;i<=n;i++) sum[i]=sum[i-1]+read();
for(int i=1;i<=m;i++)
{
now^=1;l[now]=r[now]=0;
for(int j=i;j<=n;j++)
{
while(l[now^1]<r[now^1]&&Slope(i-1,q[now^1][l[now^1]],q[now^1][l[now^1]+1])<=sum[j]*2) l[now^1]++;
int t=q[now^1][l[now^1]];
f[i][j]=f[i-1][t]+Sqr(sum[j]-sum[t]);
while(l[now]<r[now]&&Slope(i,q[now][r[now]],j)<Slope(i,q[now][r[now]-1],q[now][r[now]])) r[now]--;
q[now][++r[now]]=j;
}
}
printf("%lld\n",m*f[m][n]-Sqr(sum[n]));
return 0;
}