【DP】APIO 2014 序列分割

Description

    小H最近迷上了一个分隔序列的游戏。在这个游戏里，小H需要将一个长度为n的非负整数序列分割成k+1个非空的子序列。为了得到k+1个子序列，小H需要重复k次以下的步骤：
    1.小H首先选择一个长度超过1的序列（一开始小H只有一个长度为n的序列——也就是一开始得到的整个序列）；
    2.选择一个位置，并通过这个位置将这个序列分割成连续的两个非空的新序列。
    每次进行上述步骤之后，小H将会得到一定的分数。这个分数为两个新序列中元素和的乘积。小H希望选择一种最佳的分割方式，使得k轮之后，小H的总得分最大。

Input

    输入第一行包含两个整数n，k（k+1≤n）。
    第二行包含n个非负整数a1，a2，…，an（0≤ai≤10^4），表示一开始小H得到的序列。

Output

    输出第一行包含一个整数，为小H可以得到的最大分数。

Sample Input

7 3 
4 1 3 4 0 2 3 

Sample Output

108 

I think

    斜率优化DP
    假设最后分割的结果是和为a,b,c的三部分，那么由两种不同的分割顺序得分分别为：

$$res_1=a*(b+c)+b*c,\ \ res_2=c*(a+b)+a*b$$
    因为 $res_1=res_2$，所以得证分割的顺序不影响最终的结果。
    用f[i][j]表示前i个数分割为j个序列的最大分数,s[i]为前缀和，则有状态转移方程: $$f[i][j]=max\{f[k][j-1]+(s[i]-s[k])*s[k]\}$$
    若对于f[i][j],k1>k2且k1优于k2,则有
$$f[k1][j-1]+(s[i]-s[k1])*s[k1]>f[k2][j-1]+(s[i]-s[k2])*s[k2]$$
    移项(注意变号)，变形后有：
$$((s[k1]^2-f[k1][j-1])-(s[k2]^2-f[k2][j-1]))/(s[k1]-s[k2])<s[i]$$
    记左边部分为H(k1,k2)，H(k1,k2) < s[i]，那么k1优于k2，这个可以用于更新队首。若有H(k1,k2) < H(k2,k3),通过分类讨论可得k2无论如何没有k1或k3优，这个可以用于插入当前i，更新队尾。
    于是就可以用队列上斜率优化啦，空间上可以优化一维，注意在更新时为了避免用当前更新值再次更新，应当调整循环顺序或再开一个数组。而经过实践，相比于再开一个数组，不如加一个f[][2]，用位运算优化更快。

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long ll;

const int sm = 1e5+5;

int n,k,l,r,j=1,ind=0;
int x[sm],q[sm];
ll f[sm][2],s[sm];
char ch;

void read(int &x) {
    x=0;ch=getchar();
    while(ch>'9'||ch<'0')ch=getchar();
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();  
}

ll cross(int a,int b) {
    return ll((s[a]*s[a]-f[a][ind])-(s[b]*s[b]-f[b][ind]));
}

void dp(int x) {
    l=1,r=0;
    for(int i=x;i<=n;++i) {
        while(l<r&&cross(q[l+1],q[l])<s[i]*(s[q[l+1]]-s[q[l]]))++l;
        f[i][ind^1]=f[q[l]][ind]+(s[i]-s[q[l]])*s[q[l]];
        while(l<r&&cross(i,q[r])*(s[q[r]]-s[q[r-1]])<cross(q[r],q[r-1])*(s[i]-s[q[r]]))--r;
        q[++r]=i;
    }
    ind^=1;
}

int main() {
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;++i) {
        read(x[i]);
        if(x[i])s[j]=s[j-1]+x[i],++j;
    }
    n=j-1;l=r=1;
    for(int i=1;i<=k;++i)dp(i);
    printf("%lld\n",f[n][ind]);
    return 0;
}