【题解】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$，所以在对三个序列进行分割时没有顺序区别
推广到整个序列，在确定整个序列的最优解切割位置的情况下，切割顺序不影响答案

得到斜率式:

$$(sum[x]^2-sum[y]^2)-f[le xor 1][x]+f[lexor 1][y])\over sum[x]-sum[y]$$

代码

#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[2][N],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[le^1][x]+f[le^1][y])/(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[le][i]=f[le^1][q[fr]]+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[le][n]);
    rg int i=n,j=k+1;
    while(--j){i=path[j][i];printf("%d ",i);}
    return 0;
}