C++ 单调队列 || 单调队列模版题:滑动窗口

86 篇文章 3 订阅
2 篇文章 0 订阅

给定一个大小为 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

#include <iostream>
#include <cstdio>
using namespace std;

const int N = 10000010;
int q[N], hh, tt = -1;
int n, k;
int a[N];

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

    // 处理滑动窗口的最小值问题
    for(int i = 0; i < n; i ++ )
    {
        // 维护单调队列,保持队列中的元素是递增的,并且队列中的元素在窗口内
        if(hh <= tt && i - k + 1 > q[hh])
            hh ++;
        
        // 保持队列的递增性质,出队元素直到队列为空或者队尾元素大于当前元素 a[i]
        while(hh <= tt && a[q[tt]] >= a[i])
            tt --;
        
        // 入队当前元素下标 i
        q[ ++ tt] = i;
        
        // 输出当前窗口的最小值(队首元素),i 达到窗口大小 k 时开始输出
        if(i >= k - 1)
            printf("%d  ", a[q[hh]]);
    }
    puts("");
    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]]);
    }


    return 0;

}

注意q中存的是下标。

  • 27
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
单调队列(Monotonic Queue)是一种数据结构,它可以用来解决一类滑动窗口滑动窗口是指在一个长度为n的数组中,长度为k的窗口从左往右移动,每次移动一位,求出每个窗口的最大值或最小值。 单调队列的思想就是维护一个单调递增或递减的队列队列中的元素从队头到队尾是单调递增或递减的。当窗口向右移动时,先将队头元素出队,然后将新的元素从队尾入队,保持队列的单调性。这样就可以实现每个窗口的最大值或最小值的求解。 下面是单调队列求解滑动窗口最大值的C++代码: ```c++ vector<int> maxSlidingWindow(vector<int>& nums, int k) { vector<int> res; deque<int> dq; for (int i = 0; i < nums.size(); i++) { // 如果队列中的元素个数超过了窗口大小,就将队头元素出队 if (!dq.empty() && dq.front() == i - k) { dq.pop_front(); } // 将队列中比要加入的元素小的元素全部出队 while (!dq.empty() && nums[dq.back()] < nums[i]) { dq.pop_back(); } // 将新元素加入队列 dq.push_back(i); // 取当前窗口的最大值 if (i >= k - 1) { res.push_back(nums[dq.front()]); } } return res; } ``` 在上述代码中,使用双端队列deque来实现单调队列。dq.front()表示队头元素的下标,dq.back()表示队尾元素的下标。当队列为空时,dq.front()和dq.back()都是未定义的。 时间复杂度:O(n),每个元素最多被插入和删除一次。 空间复杂度:O(k),队列中最多存储k个元素。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值