bzoj3675 [Apio2014]序列分割

Description

---

给定一个序列，求把序列分成k组后的最大价值
定义分割一次的价值为两个序列元素和的乘积

Solution

---

不难想到分割的顺序并不影响
设f[i,j]表示前i个元素分割了j次的最大价值，枚举一个分割位置k
可以发现题目给出的价值计算并不好做，因此可以只统计这一段序列的贡献，这样每一段的贡献都是独立互不影响的了
二维的dp可以固定j，然后上斜率优化即可，注意内存问题要滚和LL问题

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef long long LL;
const int N=100005;

LL f[2][N],s[N];
int q[N],a[N];

double slope(int j,int k,int l) {
    double ret=f[(j-1)&1][k]-f[(j-1)&1][l]-s[k]*s[k]+s[l]*s[l];
    ret=ret/(s[l]-s[k]);
    return ret;
}

int main(void) {
    int n,m; scanf("%d%d",&n,&m);
    rep(i,1,n) {
        int x; scanf("%d",&x);
        if (x) a[++a[0]]=x;
    }
    rep(i,1,a[0]) s[i]=s[i-1]+a[i];
    rep(j,1,m) {
        int head=1,tail=0;
        rep(i,j,a[0]) {
            while (head<tail&&slope(j,q[tail-1],q[tail])>slope(j,q[tail],i)) tail--;
            q[++tail]=i;
            while (head<tail&&slope(j,q[head],q[head+1])<=s[i]) head++;
            int k=q[head];
            f[j&1][i]=f[(j-1)&1][k]+s[k]*(s[i]-s[k]);
        }
    }
    printf("%lld\n", f[m&1][a[0]]);
    return 0;
}