题面
【题目描述】
在一年前赢得了小镇的最佳草坪比赛后,Farm John变得很懒,再也没有修剪过草坪。现在,新一轮的最佳草坪比赛又开始了,Farm John希望能够再次夺冠。
然而,Farm John的草坪非常脏乱,因此,Farm John只能够让他的奶牛来完成这项工作。Farm John有N(1 <= N <= 100,000)只排成一排的奶牛,编号为1…N。每只奶牛的效率是不同的,奶牛i的效率为E_i(0 <= E_i <= 1,000,000,000)。
靠近的奶牛们很熟悉,因此,如果Farm John安排超过K只连续的奶牛,那么,这些奶牛就会罢工去开派对:)。因此,现在Farm John需要你的帮助,计算FJ可以得到的最大效率,并且该方案中没有连续的超过K只奶牛。
【输入】
第一行:空格隔开的两个整数 N 和 K
第二到 N+1 行:第 i+1 行有一个整数 E_i
【输出】
对第一行:一个值,表示 Farm John 可以得到的最大的效率值
【样例输入】
5 2
1
2
3
4
5
【样例输出】
12
分析:
前置知识:单调队列优化dp
显然,此题是一道dp。
设
s
u
m
[
i
]
sum[i]
sum[i]表示前i只奶牛效率之和,
d
p
[
i
]
dp[i]
dp[i]表示前i只奶牛的最大效率
可推出转移方程为:
d
p
[
i
]
=
m
a
x
(
d
p
[
j
−
1
]
+
s
u
m
[
i
]
−
s
u
m
[
j
]
)
(
i
−
k
<
=
j
<
=
i
)
dp[i]=max(dp[j-1]+sum[i]-sum[j])(i-k<=j<=i)
dp[i]=max(dp[j−1]+sum[i]−sum[j])(i−k<=j<=i)
即
:
d
p
[
i
]
=
m
a
x
(
d
p
[
j
−
1
]
−
s
u
m
[
j
]
)
+
s
u
m
[
i
]
(
i
−
k
<
=
j
<
=
i
)
即:dp[i]=max(dp[j-1]-sum[j])+sum[i](i-k<=j<=i)
即:dp[i]=max(dp[j−1]−sum[j])+sum[i](i−k<=j<=i)
令
k
[
j
]
=
d
p
[
j
−
1
]
−
s
u
m
[
j
]
;
令k[j]=dp[j-1]-sum[j];
令k[j]=dp[j−1]−sum[j];
维护一个
k
[
j
]
k[j]
k[j]的单调队列即可。
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+10;
ll n,k,l,r,q[N],dp[N],ans,s[N],x,f[N];
int main(){
scanf("%lld%lld",&n,&k);
for(int i=1;i<=n;i++){
scanf("%lld",&x);
s[i]=s[i-1]+x;
}
l=0;r=0;
q[r]=0,f[r]=0;
for(int i=1;i<=n;i++){
while(l<r&&i-q[l]>k) l++;
while(l<=r&&f[r]<dp[i-1]-s[i]) r--;
q[++r]=i;
f[r]=dp[i-1]-s[i];
dp[i]=max(dp[i],f[l]+s[i]);
}
printf("%lld",dp[n]);
return 0;
}