单调队列应用场景
单调队列的典型应用场景:
- 滑动窗口求最值。例Acwing 154 滑动窗口
单调队列思想
以单调增队列为例,在元素入队之前先将其与队尾元素相比较:
- 若元素大于队尾元素,则入队
- 若元素小于队尾元素,则删除队尾元素。重复1,2步
核心代码
for (int i = 0; i < n; ++ i)
{
scanf("%d", &a[i]);
if (i - k + 1 > q[hh]) ++ hh; // 若队首出窗口,hh加1
// 核心代码:步骤1,2
while (hh <= tt && a[i] <= a[q[tt]]) -- tt; // 步骤2
q[++ tt] = i; // 步骤1
if (i + 1 >= k) printf("%d ", a[q[hh]]); // 输出结果
}
完整代码
// 单调队列
#include <iostream>
using namespace std;
const int N = 1e6 + 10;
int n, k;
int a[N], q[N];
int main()
{
scanf("%d%d", &n, &k);
for (int i = 0; i < n; ++i) scanf("%d", &a[i]);
int hh = 0, tt = -1;
for (int i = 0; i < n; ++i)
{
// 判断队头是否已经滑出窗口
if (hh <= tt && i - k + 1 > q[hh]) hh++;
while (hh <= tt && a[q[tt]] >= a[i]) tt--;
q[ ++tt ] = i;
if (i >= k - 1) printf("%d ", a[q[hh]]);
}
puts("");
hh = 0, tt = -1;
for (int i = 0; i < n; ++i)
{
// 核心代码:步骤1, 2
if (hh <= tt && i - k + 1 > q[hh]) hh++;
while (hh <= tt && a[q[tt]] <= a[i]) tt--;
q[ ++tt ] = i;
if (i >= k - 1) printf("%d ", a[q[hh]]);
}
puts("");
return 0;
}