洛谷 P1886 滑动窗口 /【模板】单调队列

最新推荐文章于 2024-07-19 13:59:11 发布

moyucaiji

最新推荐文章于 2024-07-19 13:59:11 发布

阅读量204

点赞数

分类专栏：洛谷文章标签：算法 c++

本文链接：https://blog.csdn.net/m0_63063592/article/details/128847081

版权

洛谷专栏收录该内容

10 篇文章 0 订阅

订阅专栏

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

题目描述

有一个长为 $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\%$ 的数据， $\le n \le 10^5$ ；
对于 $100\%$ 的数据， $1\le k \le n \le 10^6$ ， $a_i \in [-2^{31},2^{31})$ 。

解题思路

单调队列的模板

代码

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

struct node
{
    int val, pos;
};
int main()
{
    ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);

    // 区间为 [i-k,i]
    int n, k;
    cin >> n >> k;
    node num[n+1];
    deque<node>qmin, qmax;
    for(int i=1;i<=n;i++)
    {
        cin >> num[i].val;
        num[i].pos = i;
    }
    for(int i=1;i<=n;i++)
    {
        while(!qmin.empty() && qmin.back().val > num[i].val) qmin.pop_back();
        qmin.push_back(num[i]);
        if(qmin.front().pos == i-k) qmin.pop_front();
        if(i>=k) cout << qmin.front().val << " ";
    }
    cout << endl;
    for(int i=1;i<=n;i++)
    {
        while(!qmax.empty() && qmax.back().val < num[i].val) qmax.pop_back();
        qmax.push_back(num[i]);
        if(qmax.front().pos == i-k) qmax.pop_front();
        if(i>=k) cout << qmax.front().val << " ";
    }
    return 0;
}