BZOJ 3675 序列分割

Description

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

Input

输入文件的第一行包含两个整数n和尼(k+1≤n)。
第二行包含n个非负整数a1，n2…．，an(0≤ai≤10^4)，表示一开始小H得
到的序列。

Output

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

Sample Input

7 3
4 1 3 4 0 2 3

Sample Output

108

HINT

【样例说明】

在样例中，小H可以通过如下3轮操作得到108分：

1．-开始小H有一个序列(4，1，3，4，0，2，3)。小H选择在第1个数之后的位置

将序列分成两部分，并得到4×(1+3+4+0+2+3)=52分。

2．这一轮开始时小H有两个序列：(4)，(1，3，4，0，2，3)。小H选择在第3个数

字之后的位置将第二个序列分成两部分，并得到(1+3)×(4+0+2+

3)=36分。

3．这一轮开始时小H有三个序列：(4)，(1，3)，(4，0，2，3)。小H选择在第5个

数字之后的位置将第三个序列分成两部分，并得到(4+0)×(2+3)=

20分。

经过上述三轮操作，小H将会得到四个子序列：(4)，(1，3)，(4，0)，(2，3)并总共得到52+36+20=108分。

【数据规模与评分】

：数据满足2≤n≤100000,1≤k≤min(n -1，200)。

事实证明，APIO也有水题

首先我们看三个数的情况a，b，c

显然（a+b）*c+a*b=a*（b+c)+b*c

三个数时分割顺序不会对结果产生影响

由数学归纳法得n个数的分割顺序对结果并没有影响

结果只跟最终分割出来的序列形态有关

则dp【i】=max（dp【j】+（s【i】-s【j】）*s【j】）；

其中s是前缀和，已使用滚动数组

dp[i]=max(dp[j]+s[i]*s[j]-s[j]*s[j]);

设k的决策没j好

dp[k]+s[i]*s[k]-s[k]*s[k]<=dp[j]+s[i]*s[j]-s[j]*s[j]

dp[k]+s[i]*s[k]-s[k]*s[k]-dp[j]-s[i]*s[j]+s[j]*s[j]<=0

dp[k]-dp[j]-s[k]*s[k]+s[j]*s[j]<=s[i]*s[j]-s[i]*s[k]

(dp[k]-dp[j]-s[k]*s[k]+s[j]*s[j])/(s[j]-s[k])<=s[i]

可用斜率优化

代码如下：

#include<cstdio>

#include<cstdlib>

#include<algorithm>

#include<iostream>

using namespace std;

typedef long long LL;

const int maxn=100002;

int n,k,q[maxn];

LL dp[maxn][2],s[maxn],c[maxn],a[maxn],tot=0;

double slop( int k, int j){

     return ( double )(dp[k][0]-dp[j][0]-s[k]*s[k]+s[j]*s[j])/( double )(s[j]-s[k]);

}

void DP( int x){

     int h=1,t=0;

     for ( int i=x;i<=tot;++i){

         while (h<t&&slop(q[t],i-1)<slop(q[t-1],q[t]))--t;

         q[++t]=i-1;

         while (h<t&&slop(q[h],q[h+1])<s[i])++h;

         dp[i][1]=dp[q[h]][0]+(s[i]-s[q[h]])*s[q[h]];

     }

     for ( int i=x;i<=tot;++i)swap(dp[i][0],dp[i][1]);

}

int main(){

     scanf ( "%d%d" ,&n,&k);

     for ( int i=1;i<=n;++i){

         scanf ( "%lld" ,&c[i]);

     }

     for ( int i=1;i<=n;++i) if (c[i]!=0)a[++tot]=c[i];

     for ( int i=1;i<=tot;++i)s[i]=s[i-1]+a[i];

     for ( int i=2;i<=k+1;++i){

         DP(i);

     }

     printf ( "%lld\n" ,dp[tot][0]);

     return 0;

}