根据乘法分配律,其实最后答案就是分割后,对每个块的和,两两求乘积加和。
那么割的顺序就没有影响了。
递推式可以写成
f[i][k]=min(f[j][k−1]+(sum[i]−sum[j])∗sum[j])
那么对于k>j且决策k优于决策j
f[k][l−1]+(sum[i]−sum[k])∗sum[k]>f[j][l−1]+(sum[i]−sum[j])∗sum[j]
sum[i]>f[j][l−1]−f[k][l−1]+sum[k]²−sum[j]²sum[k]−sum[j]
因为a中可能有零,而0对答案没有任何贡献和影响,为了后面方便,直接去掉就好了
显然具有决策单调性,就可以斜率优化啦!
然而这个题卡空间!需要滚动数组优化一下。
外层循环从小到大枚举分割的次数k,内层维护每个位置i,分割k次的最大值。
斜率优化搞一搞就好了
【代码】
#include <iostream>
#include <cstdio>
#include <algorithm>
#define INF 1000000001
#define N 100005
using namespace std;
typedef long long ll;
int read()
{
int 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,l,r,now;
int a[N],sum[N],q[N];
ll f[N][2];
double slope(int j,int k){
return (double)(f[j][now^1]-f[k][now^1]+(ll)sum[k]*sum[k]-(ll)sum[j]*sum[j])/(sum[k]-sum[j]);
}
int main()
{
n=read();m=read();
for(int i=1;i<=n;i++) a[i]=read();
int top=0;
for(int i=1;i<=n;i++) if(a[i]) a[++top]=a[i];
n=top;for(int i=1;i<=n;i++) sum[i]=sum[i-1]+a[i];
for(int k=1;k<=m;k++)
{
r=l=0;now^=1;
for(int i=k;i<=n;i++)
{
while(l<r&&slope(q[l],q[l+1])<sum[i]) l++;
int t=q[l];
f[i][now]=f[t][now^1]+(ll)(sum[i]-sum[t])*sum[t];
while(l<r&&slope(q[r],i)<slope(q[r-1],q[r])) r--;q[++r]=i;
}
}
printf("%lld\n",f[n][now]);
return 0;
}