【P1886】 滑动窗口 单调队列

题目描述

有一个长为 n 的序列 a以及一个大小为 k 的窗口。现在这个从左边开始向右滑动，每次滑动一个单位，求出每次滑动后窗口中的最大值和最小值。

链接

传送门

代码

这里用了用双端队列实现的单调队列，以维护最大值为例。新的元素与队尾相比如果大就让队尾处队，同时我们还需要考虑窗口大小的限制，如果队尾和队首序号差大于等于k的话就就要缩小窗口，即让队首出队，这样就可以维护大小为k的窗口的最大值，时间复杂度是O(n),用双端队列实现也比较简单，在这个特定问题上的确要比线段树要更优。

#include <iostream>
#include <cstdio>
#include <deque>

using namespace std;
const int N = 1e6 + 10;
int n, k;
struct num{
    int order, val;
    num(int order = 0, int val = 0): order(order), val(val){};
};
deque<num> que1;
deque<num> que2;

int maxx[N], cnt = 0;
int main() {
    scanf("%d%d", &n, &k);
    num temp;
    for (int i = 1; i <= n; ++i) {
        temp.order = i;
        scanf("%d", &temp.val);
        while (!que1.empty() && temp.val >= que1.back().val)
            que1.pop_back();
        while (!que2.empty() && temp.val <= que2.back().val)
            que2.pop_back();

        que1.push_back(temp);
        que2.push_back(temp);

        while (!que1.empty() && i - que1.front().order >= k)
            que1.pop_front();
        while (!que2.empty() && i - que2.front().order >= k)
            que2.pop_front();

        if (i >= k){
            printf("%d ", que2.front().val);
            maxx[++cnt] = que1.front().val;
        }
    }

    printf("\n");
    for (int i = 1; i <= cnt; ++i) {
        printf("%d ", maxx[i]);
    }
    return 0;
}