单调队列
适用范围
适用于求一定区间中的最大值或者最小值,且这个区间会一直移动,此时使用单调队列可以将时间复杂度由O(mn)降低为O(n);m为区间大小,n为数组长度。
下面由求滑动窗口中的最小值为例
尤其要注意的是,单调队列中存储的并不是值,而是下标。
当一个元素入队之前,如果此元素小于队尾的元素,那么当这个元素在窗口中时,最小值永远不可能是队尾的元素,而且由于此元素的位置一定在队尾元素的右边故队尾元素永远不会被用到,所以将其出队,重复这个操作,直到队列为空,或者遇到队尾元素比这个值小的时候将其入队。
问题描述
给定一个大小为n≤106的数组。
有一个大小为k的滑动窗口,它从数组的最左边移动到最右边。
您只能在窗口中看到k个数字。
每次滑动窗口向右移动一个位置。
以下是一个例子:
该数组为[1 3 -1 -3 5 3 6 7],k为3。
窗口位置 最小值 最大值
[1 3 -1] -3 5 3 6 7 -1 3
1 [3 -1 -3] 5 3 6 7 -3 3
1 3 [-1 -3 5] 3 6 7 -3 5
1 3 -1 [-3 5 3] 6 7 -3 5
1 3 -1 -3 [5 3 6] 7 3 6
1 3 -1 -3 5 [3 6 7] 3 7
您的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值。
输入格式
输入包含两行。
第一行包含两个整数n和k,分别代表数组长度和滑动窗口的长度。
第二行有n个整数,代表数组的具体数值。
同行数据之间用空格隔开。
输出格式
输出包含两个。
第一行输出,从左至右,每个位置滑动窗口中的最小值。
第二行输出,从左至右,每个位置滑动窗口中的最大值。
代码
#include<iostream>
using namespace std;
int list[1000010];
int he = 0, en = 0;
int a[1000010];
void insert(int x){
while(1){
if(he == en){
list[en] = x;
en++;
return;
}else if(a[x] < a[list[en - 1]]){
en--;
}else{
list[en] = x;
en++;
return;
}
}
}
int main()
{
int n, k;
cin >> n >> k;
for(int i = 0; i < n; ++i){
scanf("%d", &a[i]);
}
for(int i = 0; i < n; ++i){
if(he < en && list[he] < i - k + 1){ he++;}
insert(i);
if(i >= k - 1){
cout << a[list[he]] << " ";
}
}
cout << endl;
int he = 0, ed = 0;
for(int i = 0; i < n; ++i){
if(he < ed && list[he] < i-k+1){
he++;
}
while(1){
if(he == ed){
list[ed] = i;
ed++;
break;
}else if(a[list[ed-1]] < a[i]){
ed--;
}else{
list[ed] = i;
ed++;
break;
}
}
if(i >= k - 1){
cout << a[list[he]] << " ";
}
}
return 0;
}