洛谷 P2627 修剪草坪 题解

题目：P2627 修剪草坪

单调队列 - DP - 前缀和

题目大意

给出一个长度为 $n$ 的序列，你可以取若干个数，但要求最多只能连续取 $k$ 个数，求最大能取出的数的总和

解题思路

$n\le 10^5$，想要快速求和，首选前缀和

考虑 $f[i][0]$ 表示前 $i$ 个数，第 $i$ 个数不取时的最优答案
$f[i][1]$ 就为第 $i$ 个数取时的最优答案

上面这个状态，可以很轻松的想到 $O(n^2)$ 的转移：
$f[i][0]=max(f[i-1][0],f[i-1][1])$，此部分 $O(1)$ 就可以处理完
f [ i ] [ 1 ] = m a x { f [ j ] [ 0 ] + ∑ k = j + 1 i a [ i ] } f[i][1]=max \{ f[j][0]+\sum_{k=j+1}^{i} a[i] \} f[i][1]=max{f[j][0]+∑k=j+1i​a[i]}，$(i-k\le j\le i-1)$ 也就是：
f [ i ] [ 1 ] = m a x { f [ j ] [ 0 ] + s [ i ] − s [ j ] } f[i][1]=max \{ f[j][0]+s[i]-s[j] \} f[i][1]=max{f[j][0]+s[i]−s[j]}
f [ i ] [ 1 ] = m a x { f [ j ] [ 0 ] − s [ j ] } + s [ i ] f[i][1]=max \{ f[j][0]-s[j] \} +s[i] f[i][1]=max{f[j][0]−s[j]}+s[i] （将$s[i]$提出来）

我们上面将 $s[i]$ 提出后，就变成了求 $f[j][0]-s[j]$ 的最大值，最后加上 $s[i]$。这里我们就可用单调队列来维护最大值啦！

十年OI一场空，不开long long见祖宗

AC代码：

#include<cstdio>
#include<iostream>
#include<deque>
using namespace std;
const long long Maxn=100000+20,inf=0x3f3f3f3f;
long long a[Maxn],s[Maxn],f[Maxn][2];
deque <long long> q;
long long n,k;
inline long long read()
{
	long long s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0' && ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
	return s*w;
}
void pop(long long x)
{
	while(1)
	{
		if(q.empty())break;
		if(q.front()>=x-k)break;
		q.pop_front();
	}
}
void push(long long x)
{
	while(1)
	{
		if(q.empty())break;
		long long i=q.back();
		if(f[i][0]-s[i]>=f[x][0]-s[x] && i!=x)break;
		q.pop_back();
	}
	q.push_back(x);
}
int main()
{
//	freopen("in.txt","r",stdin);
	n=read(),k=read();
	for(long long i=1;i<=n;++i)
	a[i]=read(),s[i]=s[i-1]+a[i];
	
	for(long long i=1;i<=n;++i)
	{
		pop(i);
		push(i-1);
		long long j=q.front();
		f[i][1]=s[i]+f[j][0]-s[j];
		f[i][0]=max(f[i-1][1],f[i-1][0]);
	}
	printf("%lld\n",max(f[n][0],f[n][1]));
	
	return 0;
}