算法基础课:滑动窗口

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e6 + 10;

int a[N], q[N];//q[N]维护的是数组下标。对应的a[q[i]]应当随着i的增加而增加/减少

int main()
{
    int n, k;
    cin >> n >> k;
    int hh = 0, tt = -1;
    for (int i = 0; i < n; i ++ ) 
    {
        scanf("%d", &a[i]);
        if (i - k + 1 > q[hh]) hh ++;//若队首不再被包含在窗口里
        while (hh <= tt && a[q[tt]] > a[i]) tt --;//当队尾(最大值)比下一个要读入的数还要大时
        q[++ tt] = i;//读入q中
        if (i >= k - 1) printf("%d ", a[q[hh]]);
    }
    cout << endl;
    hh = 0, tt = -1;
    for (int i = 0; i < n; i ++)
    {
        if (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]]);
    }
}
上一篇:【洛谷】P1972 [SDOI2009]HH的项链 (树状数组)


下一篇:数据结构——单调队列