AcWing 1087 修剪草坪

题目描述：

在一年前赢得了小镇的最佳草坪比赛后，FJ 变得很懒，再也没有修剪过草坪。

现在，新一轮的最佳草坪比赛又开始了，FJ 希望能够再次夺冠。

然而，FJ 的草坪非常脏乱，因此，FJ 只能够让他的奶牛来完成这项工作。

FJ 有 N 只排成一排的奶牛，编号为 1 到 N。

每只奶牛的效率是不同的，奶牛 i 的效率为 Ei。

编号相邻的奶牛们很熟悉，如果 FJ 安排超过 K 只编号连续的奶牛，那么这些奶牛就会罢工去开派对。

因此，现在 FJ 需要你的帮助，找到最合理的安排方案并计算 FJ 可以得到的最大效率。

注意，方案需满足不能包含超过 K 只编号连续的奶牛。

输入格式

第一行：空格隔开的两个整数 N 和 K；

第二到 N+1 行：第 i+1 行有一个整数 Ei。

输出格式

共一行，包含一个数值，表示 FJ 可以得到的最大的效率值。

数据范围

1≤N≤10^5
0≤Ei≤10^9

输入样例：

5 2
1
2
3
4
5

输出样例：

12

样例解释

FJ 有 5 只奶牛，效率分别为 1、2、3、4、5。

FJ 希望选取的奶牛效率总和最大，但是他不能选取超过 2 只连续的奶牛。

因此可以选择第三只以外的其他奶牛，总的效率为 1 + 2 + 4 + 5 = 12。

分析：

本题与烽火传递那题看起来比较相似，但是区别还是很大的。烽火传递那题中，可以用f[i]表示在前i个烽火台中第i个发出信号的最小代价，因为要求每隔多少距离至少要有一个烽火台发出信号。而本题要求编号连续的奶牛不能超过K头，可以理解为两个未被安排的奶牛之间被被安排奶牛不能超过K头，但是我们不能直接用f[i]表示前i头牛中第i头未被安排的最大效率值，因为这样我们没法去计算第i头牛被安排了的最大效率值。本题的f[i]应该是前i头牛中能获得的最大效率值，那么第i头牛可以安排也可以不安排，不安排的话，f[i] = f[i-1]。安排第i头牛的话，就要考虑i前面还有几头编号连续的被安排的牛，这个数字可以是0,1,2，...，K - 1.。假设编号从j + 1到i的牛都被安排了，并且编号j的牛没有被安排，那么此时f[i] = f[j-1] + s[i] - s[j]，其中s[i]表示效率的前缀和。因为计算f[i]都需要考虑前面的K头牛的状态，所以可以使用单调队列优化。f[i] = s[i] + f[j-1]-s[j]，对于f[i]而言，s[i]是固定的，前面离i最近的未被安排的牛是j，我们要找其中f[j-1] - s[j]最大的一个，所以单调队列中的元素的f[j-1]-s[j]值应该自队头到队尾单调递减。

如果i = 6，m = 3，则4 5 6都可以被安排，最近不能安排的是3，也就是我们计算f[i]时，最远需要用到队列中i - m的元素，故i - q[hh] > m时就应该出队头了。此时的队头应该是没有被安排的奶牛编号，f[i] = max(f[i-1],f[q[hh]-1]+s[i]-s[q[hh]])，当然还要注意特判q[hh] - 1不能小于0,。出队尾时，同样需要f[i-1] - s[i] >= f[q[tt]-1] - s[q[tt]]才能够出队尾，来保证单调队列的性质。

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100005;
typedef long long ll;
int q[N];
ll s[N],f[N];
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= n;i++){
        scanf("%lld",&s[i]);
        s[i] += s[i - 1];
    }
    int hh = 0,tt = 0;
    for(int i = 1;i <= n;i++){
        while(i - q[hh] > m)   hh++;
        f[i] = max(f[i - 1],f[max(0,q[hh] - 1)] + s[i] - s[q[hh]]);
        while(hh <= tt && f[i - 1] - s[i] >= f[max(0,q[tt] - 1)] - s[q[tt]]) tt--;
        q[++tt] = i;
    }
    printf("%lld\n",f[n]);
    return 0;
}