单调队列
单调队列的基础是单调栈,单调队列是队尾进,队头出,单调栈则不然,只有栈顶(队尾)一个方向,但是在维护单调性的时候,和单调栈一样,在队尾处进行维护。
有关单调栈的学习请参考单调栈初步学习
单调栈和单调队列的结果也几乎都是一样的,
升序排列的单调栈,栈底Min,栈顶Max;升序排列的单调队列,队头Min,队尾Max。
代码实现
#include <bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int q[N],hh,tt;
int main()
{
int n,x;
cin>>n;
//升序排列的单调队列(不重复),队头小,队尾大
hh=0,tt=-1;
for (int i = 0; i < n; ++i) {
cin>>x;
while (hh<=tt && x<=q[tt]) tt--;
q[++tt]=x;
}
for (int i = hh; i <= tt; ++i) cout<<q[i]<<" ";
cout<<endl;
//降序排列的单调队列(不重复),队头大,队尾小
hh=0,tt=-1;
for (int i = 0; i < n; ++i) {
cin>>x;
while (hh<=tt && x>=q[tt]) tt--;
q[++tt]=x;
}
for (int i = hh; i <= tt; ++i) cout<<q[i]<<" ";
}
测试:
5
10 3 7 4 12
3 4 12
10 3 7 4 12
12
AcWing 154.滑动窗口
给定一个大小为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个整数,代表数组的具体数值。
同行数据之间用空格隔开。
输出格式
输出包含两个。
第一行输出,从左至右,每个位置滑动窗口中的最小值。
第二行输出,从左至右,每个位置滑动窗口中的最大值。
输入样例:
8 3
1 3 -1 -3 5 3 6 7
输出样例:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
代码实现:
#include <bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int q[N],hh,tt;//hh表示队头,tt表示队尾
int a[N]; //a数组存储输入的值,q数组作为窗口,存储对应a数组的下标
int main()
{
int n,k;
scanf("%d%d",&n,&k);
for (int i = 0; i < n; ++i) scanf("%d",a+i);
//寻找窗口的最小值,从队列头到队列尾升序排列
hh=0,tt=-1; //hh<tt 说明队列为空
for (int i = 0; i < n; ++i) { //利用下标判断窗口里元素的个数是否超过k
if (hh<=tt && i-q[hh]+1>k) hh++;
//hh<=tt说明队列不空;i是窗口的末位置的下标,q[hh]是窗口的初位置的下标,i-q[hh]+1表示当前窗口中元素个数,k是限定窗口中元素个数
while (hh<=tt && a[i]<=a[q[tt]]) tt--;
//等于的时候最好也弹出,因为没必要重复出现
q[++tt]=i;
if (i>=k-1) printf("%d ",a[q[hh]]);
//除过刚开始队列中元素不够k的情况,随后队列每往后移一位,都是一个新的窗口,都要输出一次,
//队列头q[hh]存储的是当前窗口值最小的下标
}
cout<<endl;
//寻址窗口的最大值,从队列头到队列尾降序排列
hh=0,tt=-1;
for (int i = 0; i < n; ++i) {
if (hh<=tt && i-q[hh]+1>k) hh++;
while (hh<=tt && a[i]>=a[q[tt]]) tt--;
q[++tt]=i;
if (i>=k-1) printf("%d ",a[q[hh]]);
//必须要保证第32行的语句再33行之前,因为可能存在经过31行的while循环后,队列被清空,tt<hh的情况:tt=hh-1(只有这一种情况)
//此时说明,a[i]是最大值,把队列里的元素都清空了,因此需要先把当前元素加入,再输出。
}
cout<<endl;
return 0;
}
遗留问题
能不能和单调栈一样,队列存储的不是下标,而是数值就完成上面的题
20201.03.13
不可,需要用队列中存储下标来维护窗口的大小为k。
if (hh<=tt && i-q[hh]+1>k)
这个理解有误,而且hh<=tt
这个判断也没必要写。
理论上,i指向窗口的末尾,i-k+1指向窗口的开始;
实际上,q[tt]指向窗口的末尾,q[hh]指向窗口的开始。
因此也可以写成if (i-k+1>q[hh])
。