Description
在一年前赢得了小镇的最佳草坪比赛后,FJ变得很懒,再也没有修剪过草坪。现在,
新一轮的最佳草坪比赛又开始了,FJ希望能够再次夺冠。
然而,FJ的草坪非常脏乱,因此,FJ只能够让他的奶牛来完成这项工作。FJ有N
(1 <= N <= 100,000)只排成一排的奶牛,编号为1...N。每只奶牛的效率是不同的,
奶牛i的效率为E_i(0 <= E_i <= 1,000,000,000)。
靠近的奶牛们很熟悉,因此,如果FJ安排超过K只连续的奶牛,那么,这些奶牛就会罢工
去开派对:)。因此,现在FJ需要你的帮助,计算FJ可以得到的最大效率,并且该方案中
没有连续的超过K只奶牛。
Input
* 第一行:空格隔开的两个整数N和K
* 第二到N+1行:第i+1行有一个整数E_i
Output
* 第一行:一个值,表示FJ可以得到的最大的效率值。
Sample Input
5 2
1
2
3
4
5
f[i] 代表选择第 i 个奶牛。能获得的最大值。
则 f[i] = max(f[i],f[j] + sum[i] - sum[j+1]);
我们可以发现, sum[i] 是不变的, 这个时候我们只要求出来 f[j] - sum[j+1] 的最大值就可以了。
这个就可以用单调队列来优化。
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+100;
long long f[N],sum[N],ans =0;
int a[N],n,m,k,q[N];
int main(){
scanf("%d%d",&n,&k);
for (int i = 1; i <= n; ++i){
scanf("%d",&a[i]);
sum[i] = sum[i-1] + a[i];
}
int l=0, r = 0;
for (int i = 1; i <= n; ++i){
if (q[l] < i-k-1) l++;
if (i <= k) f[i] = sum[i]; else
f[i] = max(f[i],f[q[l]] - sum[q[l]+1] + sum[i]);
while(l <= r && (f[i]-sum[i+1] >= f[q[r]] - sum[q[r]+1])) r--;
q[++r] = i;
ans = max(ans,f[i]);
}
printf("%lld\n",ans);
return 0;
}