测试地址:序列分割
做法:本题需要用到DP斜率优化。
本题首先要注意到一个性质:只要选定了切割位置,无论按什么顺序切结果都相同。
令
a,b,c
a
,
b
,
c
为切割后的三段区间的和,先切
ab
a
b
间和先切
bc
b
c
间的答案如下:
a(b+c)+bc=ab+ac+bc
a
(
b
+
c
)
+
b
c
=
a
b
+
a
c
+
b
c
c(a+b)+ab=ab+ac+bc
c
(
a
+
b
)
+
a
b
=
a
b
+
a
c
+
b
c
可见两种情况得到的结果相同,这样就可以类推到所有切割的情况。
令
f(i,j)
f
(
i
,
j
)
为前
j
j
个数,切割次所得到的最大结果,那么我们显然可以写出状态转移方程:
f(i,j)=max{f(i−1,k)+[sum(j)−sum(k)]sum(k)}
f
(
i
,
j
)
=
max
{
f
(
i
−
1
,
k
)
+
[
s
u
m
(
j
)
−
s
u
m
(
k
)
]
s
u
m
(
k
)
}
其中
sum(x)
s
u
m
(
x
)
为前
x
x
个数的和。那么令,
K=−sum(j)
K
=
−
s
u
m
(
j
)
,
x=sum(k)
x
=
s
u
m
(
k
)
,
y=f(i−1,k)−sum(k)2
y
=
f
(
i
−
1
,
k
)
−
s
u
m
(
k
)
2
,则有:
y=Kx+G
y
=
K
x
+
G
要使
G
G
最大,就是对所有点对维护一个上凸壳,凸壳和斜率为
K
K
的直线的切点就是我们选择的点。因为随
k
k
的增加单调递增,随
j
j
的增加单调递减,所以可以直接用单调队列维护凸壳,这题就做完了,时间复杂度为。
以下是本人代码:
#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;
}