AcWing 154. 滑动窗口（C++算法）

AcWing 154. 滑动窗口

1、题目（来源于AcWing）：
给定一个大小为n≤106的数组。

有一个大小为k的滑动窗口，它从数组的最左边移动到最右边。

您只能在窗口中看到k个数字。

每次滑动窗口向右移动一个位置。

以下是一个例子：

该数组为[1 3 -1 -3 5 3 6 7]，k为3。

您的任务是确定滑动窗口位于每个位置时，窗口中的最大值和最小值。

输入格式
输入包含两行。

第一行包含两个整数n和k，分别代表数组长度和滑动窗口的长度。

第二行有n个整数，代表数组的具体数值。

同行数据之间用空格隔开。

输入样例：
8 3
1 3 -1 -3 5 3 6 7
输出样例：
-1 -3 -3 -3 3 3
3 3 5 5 6 7

2、基本思想：
用暴力作法，需要O(nk)的时间复杂度，为了优化降低时间复杂度，可以用单调队列遍历整个数组，找最小值时将前面比新加入的数大的数都删去，每次只取队头，可以将时间复杂度降低到O(n).

3、步骤：
①用一个数组来存储输入的数，用另一个数组存储每个元素在原数组中的下标。
②分为找最大值和找最小值两步来写。

4、C++代码如下（该代码引用AcWing网站的代码）：

#include <iostream>

using namespace std;

const int N = 1000010;

int a[N], q[N];//a存储原数组，q存储数组元素下标

int main()
{
    int n, k;
    cin >> n >> k;
    for (int i = 0; i < n; i ++ ) cin >> a[i];
    
    int f = 0, r = -1;
    for (int i = 0; i < n; i ++ )//最小值
    {
        if (f <= r && i - q[f] + 1 > k) f ++ ;//这里f <= r可不写，但写上是好习惯
        
        while (f <= r && a[q[r]] >= a[i]) r -- ;
        q[++r] = i;
        
        if (i >= k - 1) cout << a[q[f]] << ' ';
    }
    
    cout << endl;
    
    f = 0, r = -1;
    for (int i = 0; i < n; i ++ )//最大值
    {
        if (f <= r && i - q[f] + 1 > k) f ++ ;//这里f <= r可不写，但写上是好习惯
        
        while (f <= r && a[q[r]] <= a[i]) r -- ;
        q[++r] = i;
        
        if (i >= k - 1) cout << a[q[f]] << ' ';
    }
    
    return 0;
}//该代码引用AcWing网站的代码

注意事项：
①q数组存储的是原数组中的下标
②注意过程中一些变量的初始值为零，有的地方要加1
③注意什么时候用队头，什么时候用队尾