【单调队列】滑动窗口

单调队列——滑动窗口

题目描述——洛谷P1886

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

例如：

The array is $[1,3,-1,-3,5,3,6,7]$, and $k=3$。

输入格式

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

输出格式

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

样例 #1

样例输入 #1

8 3
1 3 -1 -3 5 3 6 7

样例输出 #1

-1 -3 -3 -3 3 3
3 3 5 5 6 7

提示

【数据范围】
对于 $50\%$ 的数据，$1\le n\le 10^5$；
对于 $100\%$ 的数据，$1\le k\le n\le 10^6$，$a_i\in [-2^{31},2^{31})$。

---

模拟过程（以"样例1"为例）：

此处模拟过程以求"滑动窗口"最小值为例：

---

代码实现如下：

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e6 + 10;

int n, k;
int a[N], q[N];  //q[N]存储数组中元素的下标。

int main()
{
    scanf("%d%d", &n, &k);
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &a[i]);
    }

    int hh = 0, tt = -1;
    for (int i = 0; i < n; i++)  //"i"表示数组元素的下标。
    {
        //"hh <= tt"判断队列不为空。
        //判断队头是否已经滑出窗口。当前滑动窗口的大小(i - q[hh] + 1)大于我们设定的滑动窗口的大小(k)，就删除队头。
        if (hh <= tt && i - q[hh] + 1 > k)  //"q[hh]"表示当前队列的队头。
        {
            hh++;
        }
 
        //如果队尾元素大于"待入队元素"，队尾元素就一定不是当前窗口最小值，弹出该队尾元素。
        while (hh <= tt && a[q[tt]] >= a[i])
        {
            tt--;
        }

        q[++tt] = i;  //"待入队元素"进队列。
        if (i >= k - 1)  //当"滑动窗口"有三个元素时，才开始输出。
        {
            printf("%d ", a[q[hh]]);  //单调队列具有单调性(递增)，所以队头就是该"滑动窗口"的最小值。
        }
    }
    printf("\n");

    hh = 0, tt = -1;
    for (int i = 0; i < n; i++)
    {
        if (hh <= tt && i - k + 1 > q[hh])
        {
            hh++;
        }
        while (hh <= tt && a[q[tt]] <= a[i])
        {
            tt--;
        }

        q[++tt] = i;
        if (i >= k - 1)
        {
            printf("%d ", a[q[hh]]);  //单调队列具有单调性(递减)，所以队头就是该"滑动窗口"的最大值。
        }
    }

    system("pause");
    return 0;
}