题目描述
在一年前赢得了小镇的最佳草坪比赛后,FJ 变得很懒,再也没有修剪过草坪。
现在,新一轮的最佳草坪比赛又开始了,FJ 希望能够再次夺冠。
然而,FJ 的草坪非常脏乱,因此,FJ 只能够让他的奶牛来完成这项工作。
FJ 有 N N N只排成一排的奶牛,编号为 1 1 1 到 N N N。
每只奶牛的效率是不同的,奶牛 i i i 的效率为 E i E_i Ei。
编号相邻的奶牛们很熟悉,如果 FJ 安排超过 K K K 只编号连续的奶牛,那么这些奶牛就会罢工去开派对。
因此,现在 FJ 需要你的帮助,找到最合理的安排方案并计算 FJ 可以得到的最大效率。
注意,方案需满足不能包含超过 K K K 只编号连续的奶牛。
输入格式
第一行:空格隔开的两个整数 N N N 和 K K K;
第二到 N + 1 N+1 N+1 行:第 i + 1 i+1 i+1 行有一个整数 E i E_i Ei。
输出格式
共一行,包含一个数值,表示 FJ 可以得到的最大的效率值。
数据范围
1
≤
N
≤
1
0
5
1≤N≤10^5
1≤N≤105,
0
≤
E
i
≤
1
0
9
0≤E_i≤10^9
0≤Ei≤109
输入样例
5 2
1
2
3
4
5
输出样例
12
样例解释
FJ 有 5 只奶牛,效率分别为 1、2、3、4、5。
FJ 希望选取的奶牛效率总和最大,但是他不能选取超过 2 只连续的奶牛。
因此可以选择第三只以外的其他奶牛,总的效率为 1 + 2 + 4 + 5 = 12。
算法思想——单调队列优化DP
根据题目描述,求的是可以得到的最大的效率值,可以考虑使用动态规划的思想进行分析。
状态表示
f[i]
表示前i
头奶牛在合法情况下(方案满足不包含超过
K
K
K 只编号连续的奶牛),可以得到的最大效率。
状态计算
在计算f[i]
时,可以根据是否选择第i
头奶牛将所有方案划分为两种情况:
一、不选择第i
头奶牛,此时最大效率为
f
[
i
]
=
f
[
i
−
1
]
f[i] = f[i - 1]
f[i]=f[i−1]
二、选择第i
头奶牛,由于不能安排超过
K
K
K 只编号连续的奶牛,因此,又可以根据i
之前已经选择编号连续的奶牛个数将该方案继续划分为k
种情况。
不妨设编号连续的奶牛个数j
,则
1
≤
j
≤
k
1\le j \le k
1≤j≤k,如下图所示:
此时最大效率为
m
a
x
{
f
[
i
−
j
−
1
]
+
s
[
i
]
−
s
[
i
−
j
]
}
max\{f[i-j-1]+s[i] - s[i-j]\}
max{f[i−j−1]+s[i]−s[i−j]},其中
s
[
i
]
s[i]
s[i]表示前
i
i
i头奶牛的效率之和。将式子整理一下:
f
[
i
]
=
m
a
x
{
f
[
i
−
j
−
1
]
−
s
[
i
−
j
]
+
s
[
i
]
}
,
1
≤
j
≤
k
f[i] = max\{f[i-j-1]- s[i-j]+s[i] \}, 1\le j \le k
f[i]=max{f[i−j−1]−s[i−j]+s[i]},1≤j≤k
对于 f [ i ] f[i] f[i]来说, s [ i ] s[i] s[i]是一个确定的值。如果设 g [ i − j ] = f [ i − j − 1 ] − s [ i − j ] g[i-j]=f[i-j-1]-s[i-j] g[i−j]=f[i−j−1]−s[i−j],那么上式求的是在长度为 k k k的区间内、 g [ i ] g[i] g[i]的最大值,可以使用单调队列进行优化。
注意:单调队列中存储的是长度为 k k k的窗口中、能够让 g [ i ] g[i] g[i]取得最大值时奶牛的编号。
综合上述,f[i]
为情况一、二的最大值。
时间复杂度
使用单调队列求窗口最大值时间复杂度为 O ( n ) O(n) O(n)。
代码实现
#include <iostream>
#include <cstdio>
using namespace std;
const int N = 100010;
typedef long long LL;
LL f[N], s[N];
int q[N];
//求滑动窗口中的最大值
LL g(int i)
{
//特殊处理边界情况
if(i == 0) return 0;
else return f[i - 1] - s[i];
}
int main()
{
int n, k;
scanf("%d%d", &n, &k);
for(int i = 1; i <= n; i ++)
{
scanf("%lld", &s[i]);
s[i] += s[i - 1]; //处理前缀和
}
int hh = 0, tt = 0;
q[tt] = 0; //初始状态,f[0] = 0;
for(int i = 1; i <= n; i ++)
{
//处理滑出窗口的奶牛编号
if(q[hh] + k < i) hh ++;
//f[i - 1]不选择第i头奶牛时的最大效率
//g(q[hh]) + s[i]选择第i头奶牛时的最大效率
f[i] = max(f[i - 1], g(q[hh]) + s[i]);
//维护单调队列,将队尾小于当前位置的元素出队
while(hh <= tt && g(q[tt]) <= g(i)) tt --;
q[++ tt] = i;
}
printf("%lld\n", f[n]);
return 0;
}