BZOJ3675: [Apio2014]序列分割

Portal

根据乘法分配律，其实最后答案就是分割后，对每个块的和，两两求乘积加和。
那么割的顺序就没有影响了。
递推式可以写成

$$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]>\frac{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;
}