小H最近迷上了一个分割序列的游戏。在这个游戏里,小H需要将一个长
度为N的非负整数序列分割成k+l个非空的子序列。为了得到k+l个子序列,
小H将重复进行七次以下的步骤:
1.小H首先选择一个长度超过1的序列(一开始小H只有一个长度为n的
序列一一也就是一开始得到的整个序列);
2.选择一个位置,并通过这个位置将这个序列分割成连续的两个非空的新
序列。
每次进行上述步骤之后,小H将会得到一定的分数。这个分数为两个新序
列中元素和的乘积。小H希望选择一种最佳的分割方案,使得k轮(次)之后,
小H的总得分最大。
输入文件的第一行包含两个整数n和尼(k+1≤n)。
第二行包含n个非负整数a1,n2….,an(0≤ai≤10^4),表示一开始小H得
到的序列。
1 #include <iostream>
2 #include <cstring>
3 #include <cstdio>
4 using namespace std;
5 const int maxn=100010;
6 long long dp[maxn][2];
7 long long s[maxn];
8 int q[maxn],front,back;
9 int main(){
10 int n,K;
11 scanf("%d%d",&n,&K);K++;
12 for(int i=1;i<=n;i++){
13 scanf("%lld",&s[i]);
14 s[i]+=s[i-1];
15 }
16 int now,pre;
17 for(int k=2;k<=K;k++){
18 now=(k+1)%2;
19 pre=k%2;
20 front=back=1;
21 q[back++]=0;
22 for(int i=1;i<=n;i++){
23 while(front<back-1&&dp[q[front+1]][pre]-dp[q[front]][pre]+(s[q[front+1]]-s[q[front]])*s[i]-s[q[front+1]]*s[q[front+1]]+s[q[front]]*s[q[front]]>=0)
24 front++;
25 dp[i][now]=dp[q[front]][pre]+s[q[front]]*(s[i]-s[q[front]]);
26 while(front<back-1&&(dp[i][pre]-dp[q[back-1]][pre]-s[i]*s[i]+s[q[back-1]]*s[q[back-1]])*(s[q[back-2]]-s[q[back-1]])<=(dp[q[back-1]][pre]-dp[q[back-2]][pre]-s[q[back-1]]*s[q[back-1]]+s[q[back-2]]*s[q[back-2]])*(s[q[back-1]]-s[i]))
27 back--;
28 q[back++]=i;
29 }
30 }
31 printf("%lld\n",dp[n][now]);
32 }