154. 滑动窗口

最新推荐文章于 2024-10-04 20:45:38 发布

蔓蔓的学习记录

最新推荐文章于 2024-10-04 20:45:38 发布

阅读量69

点赞数

分类专栏： AcWing 文章标签：算法 c++ 图论

本文链接：https://blog.csdn.net/manerzi/article/details/130067083

版权

AcWing 专栏收录该内容

40 篇文章 0 订阅

订阅专栏

文章介绍了使用单调队列作为数据结构来解决滑动窗口问题的方法，具体应用在寻找给定大小的窗口中的最小值和最大值。代码示例展示了如何动态更新队列以保持窗口内的最小/最大元素。

摘要由CSDN通过智能技术生成

在这里插入图片描述

思路：

窗口用队列来存储，类似于单调栈，只不过不是从右往左删除，而是从左往右删除，每次将窗口内比自己大的或者相等的数从左边弹出，因为窗口往右移动的过程中，最小值不可能选取比当前值还有小的数。
例如：窗口[1,3,-1]中应该将1、3弹出

AC代码：

#include<bits/stdc++.h>
using namespace std;

const int N=1e6+10;

int n,k;
int a[N],q[N],head,tail;//队列里面存储下标
int main(){
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    //先算窗口最小值
    head=0,tail=-1;
    for(int i=0;i<n;i++){
        if(i-q[head]+1>k&&head<=tail) head++;//非空队列且队列长度大于窗口长度，窗口向右移动一格
        while(a[q[tail]]>=a[i]&&head<=tail) tail--;//将窗口内比当前大的值都删除掉
        q[++tail]=i;
        if(i>=k-1) printf("%d ",a[q[head]]);
    }
    printf("\n");
    
    //再算窗口最大值
    head=0,tail=-1;
    for(int i=0;i<n;i++){
        if(i-q[head]+1>k&&head<=tail) head++;//非空且队列长度大于窗口长度，窗口向右移动一格
        while(a[q[tail]]<=a[i]&&head<=tail) tail--;//将窗口中比当前小的都删掉
        q[++tail]=i;
        if(i>=k-1) printf("%d ",a[q[head]]);
    }
    printf("\n");
}