AcWing 154. 滑动窗口(单调队列)

思路:先求滑动窗口的最小值,对暴力做法进行优化,用一个队列维护滑动窗口,如果某个窗口内的某个位置 i 的值比前面的元素小,那么在这个窗口和后面的窗口中 i 前面的元素一定是不会被输出的,就可以删掉,这样得到的队列一定是单调递增的,队头为最小值。

如果队头元素的位置超出窗口的范围要对队列进行删减。

因为在一个窗口中最小元素的位置是不确定的,所以队列要存储元素的位置

#include<bits/stdc++.h>
using namespace std;

const int N=1000010;
int n,k;
int arr[N],q[N];

int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>arr[i];
	}
	int hh=1,tt=0;
	for(int i=1;i<=n;i++){
		if(hh<=tt&&q[hh]<=i-k){
			hh++;
		}
		while(hh<=tt&&arr[q[tt]]>=arr[i]){
			tt--;
		}
		q[++tt]=i;
		if(i>=k){
			cout<<arr[q[hh]]<<" ";
		}
	}
	cout<<endl;
	hh=1;tt=0;
	for(int i=1;i<=n;i++){
		if(hh<=tt&&q[hh]<=i-k){
			hh++;
		}
		while(hh<=tt&&arr[q[tt]]<=arr[i]){
			tt--;
		}
		q[++tt]=i;
		if(i>=k){
			cout<<arr[q[hh]]<<" ";
		}
	}
	return 0;
}
经验分享 程序员 微信小程序 职场和发展