单调队列(滑动窗口问题)

单调队列定义

队内元素是单调的,递增或递减。

单调队列性质

1、队尾既可以执行入队操作,也可以执行出队操作, 但队头只能执行出队操作

单调队列解决滑动窗口最大最小值问题

题目:

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

输入格式
输入一共有两行，第一行有两个正整数 n,k。 第二行 n 个整数，表示序列 a

输出格式
输出共两行，第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值

样例输入:
8 3
1 3 -1 -3 5 3 6 7

样例输出:
-1 -3 -3 -3 3 3
3 3 5 5 6 7

题目分析:

首先给出一个数组,有n个数字,然后给出滑动区间长度为k,每次滑动窗口右移动一个单位,每次找出窗口内的最大和最小值。

复杂度分析:

如果我用两个for循环,内存for循环找出最大值,或最小值,那么复杂度为O(nm)
如果我用两个优先队列,每次取出最大值,最小值。那么复杂度为O(nlogn)
使用单调队列,每次操作O(1),一共n次操作复杂度O(n);

代码实现

我们用一个数组q,存放队列中元素下标,然后我们用head,和tail维护单调队列!

1、队首出队:
当队首元素从窗口滑出时,队首元素出队head++

2、队尾入队:
当新元素滑入窗口时,要把新元素从队尾插入,此时分为两种情况:
(以每次滑出最大值为例)

1、直接插入:如果新元素小于队尾元素,直接插入,tail++

2、先删除后插:如果新元素大于等于队尾元素,那就先删除队尾元素,删除方法:tail–。用一个循环一直删到队尾元素比新插入的元素大为止!

完整代码:

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

int q1[1000100],q2[1000100];
int  a[1000010];
int n,m;

void min_deque()
{
    //初始化,队头在前,队尾在后
    int h=1,t=0;
    for(int i=1;i<=n;i++)
    {
        //如果队头下标,不在窗口内,队头出队
        while(h<=t&&q1[h]+m<=i) h++;
        //队尾入队
        while(h<=t&&a[i]<a[q1[t]])t--;
        q1[++t]=i;//队尾插入
        if(i>=m){
            printf("%d ",a[q1[h]]);
        }
    }
     cout<<endl;
}
void max_deque()//与min_deque相同
{
    int h=1,t=0;
    for(int i=1;i<=n;i++)
    {
        while(h<=t&&q2[h]+m<=i)h++;
        while(h<=t&&a[i]>a[q2[t]])t--;//这里换成>号
        q2[++t]=i;
        if(i>=m)
        {
            printf("%d ",a[q2[h]]);
        }
    }
}
int main()
{
   scanf("%d%d",&n,&m);
   for(int i=1;i<=n;i++)
   {
       scanf("%d",&a[i]);
   }
   min_deque();
   max_deque();
    return 0;
}