修剪草坪（单调队列优化）

最新推荐文章于 2023-02-06 15:27:58 发布

Helioca

最新推荐文章于 2023-02-06 15:27:58 发布

阅读量263

点赞数

文章标签： c++

本文链接：https://blog.csdn.net/ybC202444/article/details/127161633

版权

样例

样例输入

样例输出

复制12

鹅鹅鹅鹅鹅，颓废了很久都没写过博客了，也好久（十天）没有整天上过竞赛了。。。

题目分析：这是一道dp题，显然如果直接写的话时间复杂度为O(nk)，显然是会超时的，那么我们就需要用单调队列来优化这道题。

但是在优化之前，我们可以先看看普通超时做法

#include <bits/stdc++.h>
using namespace std ;
const int MAXN = 1e5 + 5 ;
long long n, k, a[MAXN], q[MAXN], head = 1, tail = 1 ;
long long Sum[MAXN], dp[MAXN][2] ;
long long read()
{
	long long x = 0, f = 1 ;
	char c = getchar() ;
	while(c < '0' || c > '9')
	{
		if(c == '-')	f = -1 ;
		c = getchar() ;
	}  
	while(c >= '0' && c <= '9')
	{
		x = x * 10 + c - '0' ;
		c = getchar() ;
	}
	return (x * f) ;
} 
int main()
{
	n = read(), k = read() ;
	for(long long i = 1; i <= n; i ++)
	{
		a[i] = read() ;
		Sum[i] = Sum[i - 1] + a[i] ;
	}
	for(long long i = 1; i <= n; i ++)
	{
		dp[i][0] = max(dp[i - 1][1], dp[i - 1][0]) ;
		for(long long j = max((long long) 0, i - k); j <= i - 1; j ++)
		{
			dp[i][1] = max(dp[i][1], dp[j][0] + Sum[i] - Sum[j]) ;
		}
	}
	printf("%lld\n", max(dp[n][1], dp[n][0])) ;
}

然后我们就要开始思考将内层循环优化成一个单调队列，我们提取一下未知求解的最大值应该是dp[j][0] - Sum[j]，所以我们就需要将队头元素和这个比较。

#include <bits/stdc++.h>
using namespace std ;
const int MAXN = 1e5 + 5 ;
long long n, k, a[MAXN], q[MAXN], head = 1, tail = 1 ;
long long Sum[MAXN], dp[MAXN][2] ;
long long read()
{
	long long x = 0, f = 1 ;
	char c = getchar() ;
	while(c < '0' || c > '9')
	{
		if(c == '-')	f = -1 ;
		c = getchar() ;
	}  
	while(c >= '0' && c <= '9')
	{
		x = x * 10 + c - '0' ;
		c = getchar() ;
	}
	return (x * f) ;
} 
int main()
{
	n = read(), k = read() ;
	for(long long i = 1; i <= n; i ++)
	{
		a[i] = read() ;
		Sum[i] = Sum[i - 1] + a[i] ;
	}
	for(long long i = 1; i <= n; i ++)
	{
		dp[i][0] = max(dp[i - 1][1], dp[i - 1][0]) ;
		while(q[head] < i - k && head <= tail)	head ++ ;
		dp[i][1] = dp[q[head]][0] + Sum[i] - Sum[q[head]] ;
		while(dp[i][0] - Sum[i] > dp[q[tail]][0] - Sum[q[tail]] && head <= tail)tail -- ;
		q[++tail] = i ;
	}
	printf("%lld\n", max(dp[n][1], dp[n][0])) ;
}