C - 滑动窗口 /【模板】单调队列

Description

有一个长为 n 的序列 a,以及一个大小为 k 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。

例如:

The array is [1,3,−1,−3,5,3,6,7] and k=3。

Input

输入一共有两行,第一行有两个正整数 n,k。

第二行 n 个整数,表示序列 a

Output

输出共两行,第一行为每次窗口滑动的最小值

第二行为每次窗口滑动的最大值

Sample 1

Inputcopy Outputcopy
8 3 1 3 -1 -3 5 3 6 7 -1 -3 -3 -3 3 3 3 3 5 5 6 7

Hint

【数据范围】

对于50% 的数据,1≤n≤10^5;

对于100% 的数据,1≤k≤n≤10^6,ai​∈[−231,231)。

复制代码
#include<iostream>
using namespace std;
const int N = 1e6 + 10;
int a[N], maxq[N], minq[N];
int n, m;

//h是队头,用于输出答案;t是队尾,用于加入和删除元素
//注意:对头为左,队尾为右,队头只出,队尾可出可进;因此若队内有元素,则h<=t
//max队内的元素因保持单调递减的性质,加入的不是元素值,而是元素所在数组a中的下标
void maxfind() {
	int h = 0, t = -1;

	for (int i = 1; i <= n; i++)
	{
		if (h <= t && maxq[h] <= i - m) h++;    //不在当前滑动窗口的元素删除
		while (h <= t && a[i] >= a[maxq[t]]) t--; //a[i]为当前元素,若当前元素大于队尾的元素,则需要删除,时其满足单调递减的性质
		maxq[++t] = i;   //加入新元素的下标
		if(i>=m) cout << a[maxq[h]] << " ";//输出结果
	}
}
//原理同上
void minfind() {
	int h = 0, t = -1;

	for (int i = 1; i <= n; i++)
	{
		if (h <= t && minq[h] <= i - m) h++;
		while (h <= t && a[i] <= a[minq[t]]) t--;
		minq[++t] = i;
		if(i>=m) cout << a[minq[h]] << " ";
	}
}

int main()
{
	cin >> n >> m;
	for (int i = 1; i <= n; i++) cin >> a[i];
	minfind();
	cout << endl;
	maxfind();
	return 0;
}

单调队列模板(滑动窗口)_滑动窗口 /【模板】单调队列_胡牧之.的博客-CSDN博客

相关推荐
Han.miracle1 小时前
数据结构——二叉树的从前序与中序遍历序列构造二叉树
java·数据结构·学习·算法·leetcode
北冥湖畔的燕雀2 小时前
C++泛型编程(函数模板以及类模板)
开发语言·c++
mit6.8243 小时前
前后缀分解
算法
QX_hao4 小时前
【Go】--map和struct数据类型
开发语言·后端·golang
你好,我叫C小白4 小时前
C语言 循环结构(1)
c语言·开发语言·算法·while·do...while
Evand J5 小时前
【MATLAB例程】基于USBL和DVL的线性回归误差补偿,对USBL和DVL导航数据进行相互补偿,提高定位精度,附代码下载链接
开发语言·matlab·线性回归·水下定位·usbl·dvl
朱嘉鼎6 小时前
状态机的介绍
c语言·单片机
寂静山林6 小时前
UVa 10228 A Star not a Tree?
算法
爱喝白开水a6 小时前
LangChain 基础系列之 Prompt 工程详解:从设计原理到实战模板_langchain prompt
开发语言·数据库·人工智能·python·langchain·prompt·知识图谱
Neverfadeaway6 小时前
【C语言】深入理解函数指针数组应用(4)
c语言·开发语言·算法·回调函数·转移表·c语言实现计算器