Acwing算法基础篇(二) 滑动窗口(单调队列)

给定一个大小为 n≤106 的数组。

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

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

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

以下是一个例子:

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

窗口位置	最小值	最大值
[1 3 -1] -3 5 3 6 7	-1	3
1 [3 -1 -3] 5 3 6 7	-3	3
1 3 [-1 -3 5] 3 6 7	-3	5
1 3 -1 [-3 5 3] 6 7	-3	5
1 3 -1 -3 [5 3 6] 7	3	6
1 3 -1 -3 5 [3 6 7]	3	7
你的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值。

输入格式
输入包含两行。

第一行包含两个整数 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

思路:

 这道题本质是一个单调队列的问题,窗口其实就是一个不断有元素出入的队列。关于单调队列的思想与单调栈相似,下面举例说明。
在这里插入图片描述
 对于-1来说,它在-3的“前面”,即它会先于-3出队,同时它也大于-3,因此,当当前窗口包含-1与-3时,我们并不需要考虑-1。仅针对查找最小元素,我们秉持上面的思想,尝试在窗口的移动过程中构造这么一个队列,它里面的元素一定是单调递增,队头元素最小,队尾元素最大。随着窗口的移动,一旦新进队列的元素小于队尾元素,那么我们逐个让队尾元素出队(y总原话,但这让我颇为难受,毕竟违反了传统队列的定义),相当于上面无视-1的操作,直到新元素大于当前的队尾元素,整个队列再次单调递增。而此时寻找当前窗口下的最小值
 求最小值的做法明了,求最大值也清楚了。就是维护一个单调递减的队列,队头元素最大,队尾元素最小。一旦出现新元素大于队尾元素,那么逐个让队尾元素出队,直到它小于新的队尾元素。

代码

# include <iostream>
using namespace std;
const int N = 1000010;
//q为单调队列,a表示原数组,hh为队头,tt为队尾
int a[N], q[N], hh, tt = -1;

int main()
{
    int n, k;
    cin >> n >> k;
    for (int i = 0; i < n; ++ i)
    {
        scanf("%d", &a[i]);
        if (i - k + 1 > q[hh]) ++ hh;                  // 若队首出窗口,hh加1
        while (hh <= tt && a[i] <= a[q[tt]]) -- tt;    // 若队尾不单调,tt减1,直到新元素大于队尾元素
        q[++ tt] = i;                                  // 下标加到队尾
        if (i + 1 >= k) printf("%d ", a[q[hh]]);       // 输出结果
    }
    cout << endl;
    hh = 0; tt = -1;                                   // 重置!
    for (int i = 0; i < n; ++ i)
    {
        if (i - k + 1 > q[hh]) ++ hh;
        while (hh <= tt && a[i] >= a[q[tt]]) -- tt;
        q[++ tt] = i;
        if (i + 1 >= k) printf("%d ", a[q[hh]]);
    }
    return 0;
}

作者:Alicia编程果果
链接:https://www.acwing.com/solution/content/6564/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值