【斜率优化】BZOJ3675(Apio2014)[序列分割]题解

题目概述

有一个序列 $\{a_n\}$ ，可以分割 $m$ 次，每次分割的得分是左边总和 $\times$ 右边总和，分割之后就成为了两个新序列，求最大得分。

解题报告

显然是二维DP，定义 $f[j][i]$ 表示切了 $j$ 次，第 $j$ 次切在 $i$ 的最大得分，那么：

$f[j][i]=max\{f[j-1][k]+(sum[i]-sum[k])(sum[n]-sum[k])|k<i\}$

发现可以用斜率优化，只不过需要 $m$ 个单调队列。推荐推式子的时候不要拆开后面，这样更容易：

$\frac{f[j][i]-f[j][k]}{Sum(i)-Sum(k)}>Sum(n)-Sum(i)$

由于空间卡的比较紧，所以使用滚动数组。

示例程序

#include<cstdio>
using namespace std;
typedef long long LL;
const int maxn=1e5,maxm=200;

int n,m,Head[2],Tail[2],que[2][maxn+5];
LL sum[maxn+5],f[2][maxn+5];

#define X(x,y) (sum[y]-sum[x])
#define Y(x,y) (f[j][y]-f[j][x])
inline int Max(int j,int i)
{
    int &H=Head[j],&T=Tail[j],*Q=que[j];
    while (H<T&&Y(Q[H],Q[H+1])>=X(Q[H],Q[H+1])*(sum[n]-sum[i])) H++;
    return Q[H];
}
inline void Push(int j,int i)
{
    int &H=Head[j],&T=Tail[j],*Q=que[j];
    while (H<T&&Y(Q[T-1],Q[T])*X(Q[T],i)<=Y(Q[T],i)*X(Q[T-1],Q[T])) T--;
    Q[++T]=i;
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    scanf("%d%d",&n,&m);
    for (int i=1,x;i<=n;i++) scanf("%d",&x),sum[i]=sum[i-1]+x;
    Head[0]=1;que[0][Tail[0]=1]=0;
    for (int j=1,cur=1;j<=m;j++,cur^=1)
    {
        Head[cur]=1;Tail[cur]=0;
        for (int i=j,k;i<n;i++)
            k=Max(cur^1,i),f[cur][i]=f[cur^1][k]+X(k,i)*X(i,n),Push(cur,i);
    }
    LL ans=0;for (int i=m;i<n;i++) if (f[m&1][i]>ans) ans=f[m&1][i];
    return printf("%lld\n",ans),0;
}