测试地址:序列分割
做法:本题需要用到DP斜率优化。
本题首先要注意到一个性质:只要选定了切割位置,无论按什么顺序切结果都相同。
令为切割后的三段区间的和,先切间和先切间的答案如下:
可见两种情况得到的结果相同,这样就可以类推到所有切割的情况。
令为前个数,切割次所得到的最大结果,那么我们显然可以写出状态转移方程:
其中为前个数的和。那么令,,,,则有:
要使最大,就是对所有点对维护一个上凸壳,凸壳和斜率为的直线的切点就是我们选择的点。因为随的增加单调递增,随的增加单调递减,所以可以直接用单调队列维护凸壳,这题就做完了,时间复杂度为。
以下是本人代码:
#include <bits/stdc++.h>
#define ll long long
using namespace std;
int n,k,q[100010],h,t;
ll a[100010],sum[100010],f[2][100010];
ll x[100010],y[100010];
bool larger(ll x1,ll y1,ll x2,ll y2)
{
return x2*y1>=x1*y2;
}
int main()
{
scanf("%d%d",&n,&k);
sum[0]=0;
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
sum[i]=sum[i-1]+a[i];
f[0][i]=0;
}
int past=0,now=1;
for(int i=1;i<=k;i++)
{
h=t=1;
q[1]=i;
x[1]=sum[i];
y[1]=f[past][i]-sum[i]*sum[i];
for(int j=i+1;j<=n;j++)
{
ll K=-sum[j],X=sum[j],Y=f[past][j]-sum[j]*sum[j];
while (h<t&&larger(x[h+1]-x[h],y[h+1]-y[h],1,K)) h++;
f[now][j]=f[past][q[h]]+(sum[j]-sum[q[h]])*sum[q[h]];
while (h<t&&larger(X-x[t],Y-y[t],x[t]-x[t-1],y[t]-y[t-1])) t--;
q[++t]=j;
x[t]=X,y[t]=Y;
}
swap(now,past);
}
printf("%lld",f[past][n]);
return 0;
}