时空限制 2000ms / 512MB
题目描述
你正在玩一个关于长度为 n 的非负整数序列的游戏。这个游戏中你需要把序列分成 k + 1 个非空的块。为了得到 k + 1块,你需要重复下面的操作 k次:
选择一个有超过一个元素的块(初始时你只有一块,即整个序列)
选择两个相邻元素把这个块从中间分开,得到两个非空的块。
每次操作后你将获得那两个新产生的块的元素和的乘积的分数。你想要最大化最后的总得分。
输入格式:
第一行包含两个整数 n 和 k。保证
k
+
1
≤
n
k + 1 \leq n
k+1≤n
第二行包含 n个非负整数
a
1
,
a
2
,
⋯
 
,
a
n
(
0
≤
a
i
≤
1
0
4
)
a_1, a_2, \cdots, a_n(0 \leq a_i \leq 10^4)
a1,a2,⋯,an(0≤ai≤104),表示前文所述的序列。
输出格式:
第一行输出你能获得的最大总得分。
第二行输出 k 个介于 1 到 n - 1之间的整数,表示为了使得总得分最大,你每次操作中分开两个块的位置。第 i 个整数 s_i 表示第 i 次操作将在 si 和 s i + 1 s_{i + 1} si+1之间把块分开。
如果有多种方案使得总得分最大,输出任意一种方案即可
说明
限制与约定
第一个子任务共 11 分,满足 1≤k<n≤10。
第二个子任务共 11 分,满足 1≤k<n≤50。
第三个子任务共 11 分,满足 1≤k<n≤200。
第四个子任务共 17 分,满足 2≤n≤1000,1≤k≤min{n−1,200}。
第五个子任务共 21 分,满足 2≤n≤10000,1≤k≤min{n−1,200}。
第六个子任务共 29 分,满足 2≤n≤100000,1≤k≤min{n−1,200}
题目分析
一开始还想写区间DP,然后以看数据范围。。。
好久没写斜率优化了,有分不清凸包方向了。。还被卡了好久精度。。我好菜啊+_+
d
p
[
i
]
[
k
]
=
m
i
n
(
d
p
[
j
]
[
k
−
1
]
)
+
s
u
m
[
j
]
∗
(
s
u
m
[
i
]
−
s
u
m
[
j
]
)
dp[i][k]=min(dp[j][k-1])+sum[j]*(sum[i]-sum[j])
dp[i][k]=min(dp[j][k−1])+sum[j]∗(sum[i]−sum[j])
s
u
m
[
j
]
2
−
d
p
[
j
]
[
k
−
1
]
=
s
u
m
[
i
]
∗
s
u
m
[
j
]
−
d
p
[
i
]
[
k
]
sum[j]^2-dp[j][k-1]=sum[i]*sum[j]-dp[i][k]
sum[j]2−dp[j][k−1]=sum[i]∗sum[j]−dp[i][k]
维护下凸壳
#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long lt;
typedef double dd;
#define sqr(x) ((x)*(x))
lt read()
{
lt f=1,x=0;
char ss=getchar();
while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
return x*f;
}
const int maxn=200010;
int n,K;
lt a[maxn],sum[maxn];
int ll,rr,q[maxn],pre[maxn][210];
lt dp[2][maxn];
void print(int x,int t)
{
if(t!=1) print(pre[x][t-1],t-1);
printf("%d ",x);
}
lt qans(int i,int j,int t){ return dp[(t-1)&1][j]+sum[j]*(sum[i]-sum[j]);}
dd calc(int a,int b,int t)
{
if(sum[a]==sum[b]) return -1e18;
dd tx=sum[b]-sum[a];
dd ty=(sqr(sum[b])-dp[(t-1)&1][b])-(sqr(sum[a])-dp[(t-1)&1][a]);
return ty/tx;
}
int main()
{
n=read();K=read();
for(int i=1;i<=n;++i)
a[i]=read(),sum[i]=sum[i-1]+a[i];
for(int t=1;t<=K;++t)
{
ll=rr=1;
for(int i=1;i<=n;++i)
{
while(ll<rr&&calc(q[ll],q[ll+1],t)<=sum[i]) ++ll;
dp[t&1][i]=qans(i,q[ll],t); pre[i][t]=q[ll];
while(ll<rr&&calc(q[rr-1],q[rr],t)>=calc(q[rr],i,t)) --rr;
q[++rr]=i;
}
}
printf("%lld\n",dp[K&1][n]);
print(pre[n][K],K);
return 0;
}