滑动窗口(单调队列,模板题)

题意

分别求滑动窗口中的最小值和最大值。

代码

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int N = 1000010;

int n, k;
int a[N];
int q[N];

int main()
{
    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; i ++) scanf("%d", &a[i]);
    int tt = -1, hh = 0;
    for(int i = 1; i <= n; i ++) {
        if(tt >= hh && i - k + 1 > q[hh]) hh ++;
        while(tt >= hh && a[q[tt]] >= a[i]) tt --;
        q[++ tt] = i;
        if(i >= k) printf("%d ", a[q[hh]]);
    }
    puts("");
    tt = -1, hh = 0;
    for(int i = 1; i <= n; i ++) {
        if(tt >= hh && i - k + 1 > q[hh]) hh ++;
        while(tt >= hh && a[q[tt]] <= a[i]) tt --;
        q[++ tt] = i;
        if(i >= k) printf("%d ", a[q[hh]]);
    }
    puts("");
    return 0;
}
posted @ 2021-02-28 16:14  pbc的成长之路  阅读(70)  评论(0)    收藏  举报