Luogu2627 修剪草坪

原题链接：https://www.luogu.org/problemnew/show/P2627

修剪草坪

题目描述

在一年前赢得了小镇的最佳草坪比赛后，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 可以得到的最大的效率值。

输入输出样例

输入样例#1：

5 2
1
2
3
4
5

输出样例#1：

12

题解

我们用$dp[i]$表示前$i$头牛的最优方案，显然有状态转移方程如下;
$$dp[i]=max(dp[j-1]+sum[i]-sum[j])(i-k<j<i)$$

其中，$sum[i]$为前缀和，$j$为不选的奶牛。但是这样的复杂度显然不科学，我们发现这玩意儿只跟$j$有关且我们每次只取一段区间内的最大值，所以我们可以用单调队列来优化这个$dp$。

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int M=1e5+5;
int n,p,que[M];
ll sum[M],dp[M];
void in()
{
	scanf("%d%d",&n,&p);
	for(int i=1;i<=n;++i)
	scanf("%lld",&sum[i]),sum[i]+=sum[i-1];
}
ll calc(int i){return dp[i-1]-sum[i];}
void ac()
{
	int le=0,ri=0,pos;
	for(int i=1;i<=n;++i)
	{
		while(le<ri&&calc(que[ri])<calc(i))ri--;
		que[++ri]=i;
		while(le<ri&&que[le]<i-p)le++;
		pos=que[le];
		dp[i]=dp[pos-1]+sum[i]-sum[pos];
	}
	printf("%lld",dp[n]);
}
int main()
{
	in();ac();
	return 0;
}