题意:给定一个大小为n≤1e6的数组。
有一个大小为k的滑动窗口,它从数组的最左边移动到最右边。
您只能在窗口中看到k个数字。
每次滑动窗口向右移动一个位置。
以下是一个例子:
该数组为[1 3 -1 -3 5 3 6 7],k为3。
输入样例:
8 3
1 3 -1 -3 5 3 6 7
输出样例:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
思路:单调栈和队列这种思维算法受众面比较小,主要就是用来解决像窗口滑动和最近合法数(满足某种条件)这种题的优化。而用数组模拟队列和栈的区别就只是在于队列有头有尾满足先进先出原则,而栈只有头(先进后出)。
代码实现:
#include<iostream>
using namespace std;
const int N = 1e6 + 5;
int n, k;
//用数组a记录原数列,数组q来模拟单调队列
int a[N], q[N];
int main()
{
cin >> n >> k;
for(int i = 0; i < n; i ++ )
cin >> a[i];
//求最小值
int hh = 0, tt = -1;
for(int i = 0; i < n ; i ++ ){
//如果队列的head必须在窗口内部
if(hh <= tt && q[hh] < i - k + 1) hh ++ ;
while(hh <= tt && a[q[tt]] >= a[i]) tt -- ;
q[ ++ tt] = i;
//必须满足每个窗快k个数
if(i >= k - 1) cout << a[q[hh]] << " ";
}
cout << endl;
//求最大值
hh = 0, tt = -1;
for(int i = 0; i < n; i ++ ){
if(hh <= tt && q[hh] < i - k + 1) hh ++ ;
while(hh <= tt && a[q[tt]] <= a[i]) tt -- ;
q[ ++ tt] = i;
if(i >= k - 1) cout << a[q[hh]] << " ";
}
cout << endl;
return 0;
}