滑动窗口 /【优先队列&双端队列--单调队列】

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

题目描述

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

例如，对于序列 $[1,3,-1,-3,5,3,6,7]$ 以及 $k=3$，有如下过程：

$$\begin{array}{ccc}\text{窗口位置}& \text{最小值}& \text{最大值}\\ \text{[1 3 -1] -3 5 3 6 7 }& -1& 3\\ \text{ 1 [3 -1 -3] 5 3 6 7 }& -3& 3\\ \text{ 1 3 [-1 -3 5] 3 6 7 }& -3& 5\\ \text{ 1 3 -1 [-3 5 3] 6 7 }& -3& 5\\ \text{ 1 3 -1 -3 [5 3 6] 7 }& 3& 6\\ \text{ 1 3 -1 -3 5 [3 6 7]}& 3& 7\end{array}$$

输入格式

输入一共有两行，第一行有两个正整数 $n,k$。
第二行 $n$ 个整数，表示序列 $a$

输出格式

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

样例 #1

样例输入 #1

8 3
1 3 -1 -3 5 3 6 7

样例输出 #1

-1 -3 -3 -3 3 3
3 3 5 5 6 7

提示

【数据范围】
对于 $50\%$ 的数据，$1\le n\le 10^5$；
对于 $100\%$ 的数据，$1\le k\le n\le 10^6$，$a_i\in [-2^{31},2^{31})$。

代码（优先队列）

#include<bits/stdc++.h>
using namespace std;
#define PII pair<int,int>
#define fi first
#define se second
const int N=1e6+10;
int a[N];//数组太大放全局 
signed main()
{
	ios::sync_with_stdio(false);cin.tie(0),cout.tie(0);
	int n,k;
	cin>>n>>k;
	
	for(int i=1;i<=n;i++)
	cin>>a[i];
	priority_queue<PII,vector<PII>,greater<PII > > q;//弹出最小值 
	for(int i=1,l=1;i<=n;i++)
	{
		q.push({a[i],i});
		if(i-l+1>=k)
		{
			while(q.top().se<l)//用while多次删 
			q.pop();
			cout<<q.top().fi<<" ";
			l++;
			
		 } 
	}
	cout<<endl;
	priority_queue<PII> p;
	for(int i=1,l=1;i<=n;i++)
	{
		p.push({a[i],i});
		if(i-l+1>=k)
		{
			while(p.top().se<l)
			p.pop();
			cout<<p.top().fi<<" ";
			l++;
		 } 
	}
	
}

注意

在 C++中， signed main 和 int main 的作用是相同的，都表示 main 函数的返回值为有符号整数。通常情况下，使用 int main 即可，但在某些情况下，可能需要使用 signed main ，例如：

1. 当使用了宏定义将 int 类型替换为其他类型时，为了确保 main 函数的返回值类型仍然是有符号整数，可以使用 signed main 。
2. 在一些特定的编译器或环境中，可能对 main 函数的返回值类型有特殊要求，此时需要使用 signed main 来满足要求。
   总的来说， signed main 是为了在特定情况下确保 main 函数的返回值类型为有符号整数，而在一般情况下，使用 int main 即可。

如果你使用的是 stack 或 queue ，则使用 push ；如果你使用的是 vector 或 list ，则使用 push_back 。

pair 是一个将两个值组合成一个值的类模板。 pair 的两个值可以具有不同的数据类型，并且可以分别通过 pair 的两个公有成员函数 first 和 second 来访问。

代码（双端队列）

#include<bits/stdc++.h>
using namespace std;
const int N=1000005;
int a[N];
deque<int> q;
int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	cin>>a[i];
	for(int i=1;i<=n;i++)//打印最小值
	{
		while(!q.empty()&&a[q.back()]>a[i])//来了一个更小的数，就把前面的删了
		q.pop_back();
		q.push_back(i);//都存起来，当有更小的就踢出
		if(i>=m)
		{
			while(!q.empty()&&q.front()<=i-m) q.pop_front();//将窗口外的踢出
			cout<<a[q.front()]<<" ";
		}
		
	}
	cout<<'\n';
	while(!q.empty()) q.pop_front();                                         
	for(int i=1;i<=n;i++)
	{
		while(!q.empty()&&a[q.back()]<a[i]) q.pop_back();
		q.push_back(i);
		if(i>=m)
		{
			 while(!q.empty()&&q.front()<=i-m) q.pop_front();
			 cout<<a[q.front()]<<" ";
		}
	}
	
}