【题解】APIO-2014 序列分割

【题目】bzoj3675

【题意】给定长度为n的序列$a_i$，让你将其分割k次，每次分数为分割两块的元素和的乘积，最大化得分

【题解】
这种题目在考虑转化图未果后一般可以确定是dp题

再看看数据范围：$n=1e5$就知道复杂度为$O(nlog_2n)$或$O(n)$

那么不管能不能A，先分析：

由于这道题要用Dp的话就最好保证从左往右顺序Dp
非要用树型Dp和平衡树的奏凯

蒟蒻不会树型Dp做，只会想法解决顺序问题
轻易可得：

对于整个序列的切分，可以抽象为多次在三个序列之间的切分选择

则对于任何一次决策，
设三个序列分别为$a$、$b$、$c$，元素和大小分别为$A$、$B$、$C$，$a$与$b$之间分界为$p1$，$b$与$c$之间为$p2$

若在当前已经确定了：
在$p1$和$p2$两个位置插入可得最优值

对于先切位置$p1$，得两次切割代价总和为$S_1=A(B+C)+BC=AB+BC+AC$
对于先切位置$p2$，得两次切割代价总和为$S_2=(A+B)C+AB=AB+BC+AC$

由于$S_1=S_2$，所以在对三个序列进行分割时没有顺序区别
推广到整个序列，在确定整个序列的最优解切割位置的情况下，切割顺序不影响答案

现在就不用管切割顺序问题了，怀着激动的心情写下了朴素$Dp$方程：

$$f[i][k]=max(f[i][k],f[j][k-1]+(sum[i]-sum[j])×sum[j]) \quad (j ∈[1,i),j∈N)$$
很快发现 $O(n^2)$要是能过， $APIO$就可以散伙了

本着优化$Dp$的目的，进行单调性分析：

对于当前需要更新的答案$f[i][k]$，若$j$比$r$更新更优，则：

$$f[j][k-1]+(sum[i]-sum[j])×sum[j]>f[r][k-1]+(sum[i]-sum[r])×sum[r]$$
$$\Rightarrow f[i][k-1]-f[r][k-1]>sum[j]^2-sum[r]^2+sum[i]×(sum[r]-sum[j])$$
$$\Rightarrow \frac{f[j][k-1]-f[r][k-1]+sum[r]^2-sum[j]^2}{sum[r]-sum[j]}>sum[i]$$

则Dp可以用单调队列优化

进一步发现第二维是无用的，可以用滚动数组(ノ｀Д)ノ滚掉

做完了

最后汇总一下：

状态
———$f[i][k]$表示在前$i$个数中分割成$k$列的最大得分，可以用滚动数组将第二维降为2

状态转移方程

$$f[i][k]=max(f[i][k],f[j][k-1]+(sum[i]-sum[j])×sum[j]) \quad (j ∈[1,i),j∈N)$$

斜率式

$$\frac{f[j][k-1]-f[r][k-1]+sum[r]^2-sum[j]^2}{sum[r]-sum[j]}$$

单调队列维护
———以$sum[i]$为基准斜率维护一个下凸包

时间复杂度：$O(n)$
空间复杂度：$O(n)$
蒟蒻的代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define oo (0x3f3f3f3f3f3f3f3f)
#define rg register
#define pow(x) ((x)*(x))

template <typename _Tp> inline void read(_Tp&x){
    rg char c11=getchar();x=0;rg bool booo=0;
    while(c11!='-'&&!isdigit(c11))c11=getchar();if(c11=='-')c11=getchar(),booo=1;
    while(isdigit(c11)){x=x*10+c11-'0';c11=getchar();}if(booo)x=-x;return ;
}

const int N=105000;
ll sum[N],f[N][2],n,k;
int path[201][N],q[N],fr,be;
bool le=0;
inline double slope(int x,int y){
    if(sum[x]==sum[y]) return -oo;
    return (double)(pow(sum[x])-pow(sum[y])-f[x][le^1]+f[y][le^1])/(1.0*(sum[x]-sum[y]));
}
int main(){
    read(n);read(k);
    for(rg int i=1;i<=n;++i)read(sum[i]),sum[i]+=sum[i-1];
    for(rg int p=1;p<=k;++p){
        fr=be=0,le=p&1;
        for(rg int i=1;i<=n;++i){
            while(fr<be && slope(q[fr],q[fr+1]) <= sum[i]) ++fr;
            f[i][le]=f[q[fr]][le^1]+sum[q[fr]]*(sum[i]-sum[q[fr]]);
            path[p][i]=q[fr];
            while(fr<be && slope(q[be-1],q[be]) >= slope(q[be],i)) --be;
            q[++be]=i;
        }
    }
    printf("%lld\n",f[n][le]);
    rg int i=n,j=k+1;
    while(--j){i=path[j][i];printf("%d ",i);}
    return 0;
}