【ACWing 算法基础】单调队列

一. 模板

常见模型：找出滑动窗口中的最大值/最小值
int a[N], q[N];  // a[N]存放数组的值，q[N]存放对应值的下标,将队列位置映射到数组的位置
int hh = 0, tt = -1;  // hh为队列头部，tt为队列尾部
for (int i = 0; i < n; i ++ )
{
    if (hh <= tt && check_out(q[hh])) hh ++ ;  // 判断队头是否滑出窗口
    while (hh <= tt && check(q[tt], i)) tt -- ;  // 构造单调队列
    q[ ++ tt] = i;  // 在队列中添加新值
}

二. 总结

• 解决的问题：
  

• 解析动画

• 解题思路：

1. 解决队首已经出窗口的问题;
2. 解决队尾与当前元素a[i]不满足单调性的问题;
3. 将当前元素下标加入队尾;
4. 如果满足条件则输出结果;

三. 例题

• 154. 滑动窗口
  

AC代码:

#include <iostream>

using namespace std;

const int N = 1e6;

int a[N], q[N];  // a[N]存放数组的值，q[N]存放对应值的下标

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    
    int n, k;
    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) {
        // 窗口超过k，队头元素出列
        if (hh <= tt &&  i - q[hh] + 1 > k) hh++;
        // 构造单调递增队列
        while (hh <= tt && a[q[tt]] >= a[i]) tt--;  // 将原先在队列中且比新元素小的元素移出队列
        q[++tt] =  i;  // 将新元下标素加入队列
        
        if (i >= k - 1) cout << a[q[hh]] << ' ';
    }
    cout << endl;
    
    hh = 0, tt = -1;
    for (int i = 0; i < n; ++i) {
        // 窗口超过k，队头元素出列
        if (hh <= tt &&  i - q[hh] + 1 > k) hh++;
        // 构造单调递减队列
        while (hh <= tt && a[q[tt]] <= a[i]) tt--;  // 将原先在队列中且比新元素小的元素移出队列
        q[++tt] =  i;  // 将新元下标素加入队列
        
        if (i >= k - 1) cout << a[q[hh]] << ' ';
    }
    cout << endl;
}