#2495. 滑动窗口 /【模板】单调队列

题目描述

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

数组是 ([1, 3, -1, -3, 5, 3, 6, 7]), ( k = 3 )。

输入格式

输入一共有两行:

第一行有两个正整数 ( n ) 和 ( k )。

第二行有 ( n ) 个整数,表示序列 ( a )。

输出格式

输出共两行:

第一行为每次窗口滑动的最小值。

第二行为每次窗口滑动的最大值。

样例

输入数据 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 \leq n \leq 10^5 )
  • 对于 100% 的数据,( 1 \leq k \leq n \leq 10^6 ),( a_i \in [-2^{31}, 2^{31}) )

参考代码:双端队列实现

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

constexpr int N = 1e6 + 7;

int a[N], n, k;

int main() {
    cin >> n >> k;
    for (int i = 0; i < n; i++) cin >> a[i];

    // 存的是下标
    // 我们的dq里面一定是单调
    // 你要么活得比我长
    // 要么能力比我强
    deque<int> dq;
    for (int i = 0; i < n; i++) {
        // 当前队首的这个点还存活不 --- 窗口长度k
        // 比如我当前位置是i,窗口长度是3
        // 可以存在的点是i, i - 1, i - 2
        if (!dq.empty() && i - k + 1 > dq.front()) {
            // 比如我现在的i = 6,k = 3 --- 4 5 6三个位置的数
            // 但是你的dq.frond() = 3,你队首是位置为3的元素
            // 已经过了你的时代 --- 你该下位了
            dq.pop_front();
        }
        // 我a[i]要把我前面能力没我强,活的没我久的全部干掉
        // 队列里如果活得没有a[i]久,能力没有a[i]强,a[i]在的一天他们就永无出头之日
        while (!dq.empty() && a[i] <= a[dq.back()]) dq.pop_back();
        dq.push_back(i);
        // 可以输出当前的最小值了 --- 窗口长度至少得达到k才可以开始输出
        if (i >= k - 1) cout << a[dq.front()] << " ";
    }
    cout << "\n";
    dq.clear();
    for (int i = 0; i < n; i++) {
        // 当前队首的这个点还存活不 --- 窗口长度k
        // 比如我当前位置是i,窗口长度是3
        // 可以存在的点是i, i - 1, i - 2
        if (!dq.empty() && i - k + 1 > dq.front()) {
            // 比如我现在的i = 6,k = 3 --- 4 5 6三个位置的数
            // 但是你的dq.frond() = 3,你队首是位置为3的元素
            // 已经过了你的时代 --- 你该下位了
            dq.pop_front();
        }
        // 我a[i]要把我前面能力没我强,活的没我久的全部干掉
        // 队列里如果活得没有a[i]久,能力没有a[i]强,a[i]在的一天他们就永无出头之日
        while (!dq.empty() && a[i] >= a[dq.back()]) dq.pop_back();
        dq.push_back(i);
        // 可以输出当前的最小值了 --- 窗口长度至少得达到k才可以开始输出
        if (i >= k - 1) cout << a[dq.front()] << " ";
    }
    cout << "\n";
    return 0;
}

参考代码:手写单调队列

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

constexpr int N = 1e6 + 7;

int a[N], dq[N];

int n, k, head = 0, tail = -1;

bool isNotEmpty() { return head <= tail; }

int top() { return dq[head]; }

void pop_front() { head += 1; }

int back() { return dq[tail]; }

void pop_back() { tail -= 1; }

void push(int x) { dq[++tail] = x; }

int main() {
    cin >> n >> k;
    for (int i = 0; i < n; i++) cin >> a[i];

    for (int i = 0; i < n; i++) {
        if (isNotEmpty() && i - k + 1 > top()) pop_front();
        while (isNotEmpty() && a[back()] >= a[i]) pop_back();
        push(i);
        if (i >= k - 1) cout << a[top()] << " ";
    }
    cout << "\n";
    head = 0, tail = -1;
    for (int i = 0; i < n; i++) {
        if (isNotEmpty() && i - k + 1 > top()) pop_front();
        while (isNotEmpty() && a[back()] <= a[i]) pop_back();
        push(i);
        if (i >= k - 1) cout << a[top()] << " ";
    }
    cout << "\n";
    return 0;
}
相关推荐
程序员莫小特2 小时前
老题新解|大整数加法
数据结构·c++·算法
过往入尘土4 小时前
服务端与客户端的简单链接
人工智能·python·算法·pycharm·大模型
zycoder.4 小时前
力扣面试经典150题day1第一题(lc88),第二题(lc27)
算法·leetcode·面试
蒙奇D索大4 小时前
【数据结构】考研数据结构核心考点:二叉排序树(BST)全方位详解与代码实现
数据结构·笔记·学习·考研·算法·改行学it
智驱力人工智能5 小时前
工厂抽烟检测系统 智能化安全管控新方案 加油站吸烟检测技术 吸烟行为智能监测
人工智能·算法·安全·边缘计算·抽烟检测算法·工厂抽烟检测系统·吸烟监测
程序员爱钓鱼5 小时前
Go语言实战案例——进阶与部署篇:编写Makefile自动构建Go项目
后端·算法·go
_Power_Y5 小时前
Java面试常用算法api速刷
java·算法·面试
艾醒(AiXing-w)5 小时前
大模型面试题剖析:模型微调中冷启动与热启动的概念、阶段与实例解析
人工智能·深度学习·算法·语言模型·自然语言处理
天选之女wow6 小时前
【代码随想录算法训练营——Day32】动态规划——509.斐波那契数、70.爬楼梯、746.使用最小花费爬楼梯
算法·leetcode·动态规划
红衣小蛇妖6 小时前
LeetCode-704-二分查找
java·算法·leetcode·职场和发展