[洛谷]P1714 切蛋糕 (#单调队列)

题目描述

今天是小Z的生日，同学们为他带来了一块蛋糕。这块蛋糕是一个长方体，被用不同色彩分成了N个相同的小块，每小块都有对应的幸运值。

小Z作为寿星，自然希望吃到的第一块蛋糕的幸运值总和最大，但小Z最多又只能吃M小块(M≤N)的蛋糕。

吃东西自然就不想思考了，于是小Z把这个任务扔给了学OI的你，请你帮他从这N小块中找出连续的k块蛋糕(k≤M)，使得其上的幸运值最大。

输入格式

输入文件cake.in的第一行是两个整数N,M。分别代表共有N小块蛋糕，小Z最多只能吃M小块。

第二行用空格隔开的N个整数，第i个整数Pi代表第i小块蛋糕的幸运值。

输出格式

输出文件cake.out只有一行，一个整数，为小Z能够得到的最大幸运值。

输入输出样例

输入 #1复制

5 2
1 2 3 4 5

输出 #1复制

9

输入 #2复制

6 3
1 -2 3 -4 5 -6

输出 #2复制

5

说明/提示

对20%的数据，N≤100。

对100%的数据，N≤500000，|Pi|≤500。 答案保证在2^31-1之内。

---

思路

n个数中取连续的k（k<=m）个数，输出其中所有可能的最大值。

#include <stdio.h>
#include <iostream>
#include <deque>
#define linf -2e9-7
using namespace std;
int n,m,s,sum[500001];
struct node
{
	int v,position;
}a[500001];
deque<node> dq;
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	register int i,j;
	cin>>n>>m;
	for(i=1;i<=n;i++)
	{
		cin>>a[i].v;
		sum[i]=sum[i-1]+a[i].v;//前缀和 
		a[i].position=i;
	}
	for(i=1;i<=n;i++)
	{
		while(!dq.empty() && dq.front().position<i-m)//不在区间的要出队 
			dq.pop_front();
		s=max(s,sum[i]-sum[dq.front().position]);//取最大值 
		while(!dq.empty() && sum[i]<=sum[dq.back().position])//维护单调递增的序列 
			dq.pop_back();
		dq.push_back(a[i]);//元素入队 
	}
	cout<<s<<endl;
	return 0;
}