AcWing 154. 滑动窗口


#include <bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int n,k;
//q是队列
int a[N],q[N];
int main(){
    scanf("%d%d",&n,&k);
    //初始化队列
    int hh=0,tt=-1;
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    for(int i=0;i<n;i++) {
        //判断队头是否已经滑出窗口
        //终点是i,起点是i-k+1,如果>q[hh],说明q[hh]已经滑出窗口
        //窗口每次只滑动一位,所以用if不用while
        if(hh<=tt&&i-k+1>q[hh]) hh++;
        //如果新插入的数,比队尾的数要小或者相等的话,队尾就没有用了,可以删掉
        while(hh<=tt&&a[q[tt]]>=a[i]) tt--;
        //在队列中插入新插入数的坐标
        q[++tt]=i;
        //从前k个数开始输出,不足k个数,不需要输出
        if(i>=k-1) printf("%d ",a[q[hh]]);
    }
    printf("\n");
    //初始化队列
    hh=0,tt=-1;
    for(int i=0;i<n;i++) {
        //判断队头是否已经滑出窗口
        //终点是i,起点是i-k+1,如果>q[hh],说明q[hh]已经滑出窗口
        //窗口每次只滑动一位,所以用if不用while
        if(hh<=tt&&i-k+1>q[hh]) hh++;
        //如果新插入的数,比队尾的数要大或者相等的话,队尾就没有用了,可以删掉
        while(hh<=tt&&a[q[tt]]<=a[i]) tt--;
        //在队列中插入新插入数的坐标
        q[++tt]=i;
        //从前k个数开始输出,不足k个数,不需要输出
        if(i>=k-1) printf("%d ",a[q[hh]]);
    }

    return 0;
}