单调队列：修剪草坪

题目描述

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

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

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

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

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

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

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

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

输入格式

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

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

输出格式

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

数据范围

$1\le N\le 10^5$,
$0\le E_i\le 10^9$

输入样例

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$ 只编号连续的奶牛），可以得到的最大效率。

状态计算

在计算f[i]时，可以根据是否选择第i头奶牛将所有方案划分为两种情况：

一、不选择第i头奶牛，此时最大效率为$f[i]=f[i-1]$

二、选择第i头奶牛，由于不能安排超过 $K$ 只编号连续的奶牛，因此，又可以根据i之前已经选择编号连续的奶牛个数将该方案继续划分为k种情况。

不妨设编号连续的奶牛个数j，则$1\le j\le 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]$表示前$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]$来说，$s[i]$是一个确定的值。如果设$g[i-j]=f[i-j-1]-s[i-j]$，那么上式求的是在长度为$k$的区间内、$g[i]$的最大值，可以使用单调队列进行优化。

注意：单调队列中存储的是长度为$k$的窗口中、能够让$g[i]$取得最大值时奶牛的编号。

综合上述，f[i]为情况一、二的最大值。

时间复杂度

使用单调队列求窗口最大值时间复杂度为$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;
}