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

题目

题目描述：

有一个长为 nn 的序列 aa，以及一个大小为 kk 的窗口。现在这个窗口从左边开始向右滑动，每次滑动一个单位，求出每次滑动后窗口中的最小值和最大值。

例如，对于序列 [1,3,−1,−3,5,3,6,7][1,3,-1,-3,5,3,6,7] 以及 k=3k = 3，有如下过程：

窗口位置最小值最大值[1   3  -1] -3   5   3   6   7 −13 1  [3  -1  -3]  5   3   6   7 −33 1   3 [-1  -3   5]  3   6   7 −35 1   3  -1 [-3   5   3]  6   7 −35 1   3  -1  -3  [5   3   6]  7 36 1   3  -1  -3   5  [3   6   7]37\def\arraystretch{1.2} \begin{array}{|c|c|c|}\hline \textsf{窗口位置} & \textsf{最小值} & \textsf{最大值} \\ \hline \verb![1 3 -1] -3 5 3 6 7 ! & -1 & 3 \\ \hline \verb! 1 [3 -1 -3] 5 3 6 7 ! & -3 & 3 \\ \hline \verb! 1 3 [-1 -3 5] 3 6 7 ! & -3 & 5 \\ \hline \verb! 1 3 -1 [-3 5 3] 6 7 ! & -3 & 5 \\ \hline \verb! 1 3 -1 -3 [5 3 6] 7 ! & 3 & 6 \\ \hline \verb! 1 3 -1 -3 5 [3 6 7]! & 3 & 7 \\ \hline \end{array}

输入格式：

输入一共有两行，第一行有两个正整数 n,kn,k；
第二行有 nn 个整数，表示序列 aa。

输出格式：

输出共两行，第一行为每次窗口滑动的最小值；
第二行为每次窗口滑动的最大值。

数据范围与说明：

【数据范围】
对于 50%50\% 的数据，1≤n≤1051 \le n \le 10^5；
对于 100%100\% 的数据，1≤k≤n≤1061\le k \le n \le 10^6，ai∈[−231,231)a_i \in [-2^{31},2^{31})。

输入输出样例 #1

输入：

1
2
8 3
1 3 -1 -3 5 3 6 7

输出：

1
2
-1 -3 -3 -3 3 3
3 3 5 5 6 7

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#include<bits/stdc++.h>
using namespace std;
int n,k;
deque<int>q;
int a[1000005];
int main(){
    cin>>n>>k;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    for(int i=0;i<n;i++){
        if(!q.empty()&&q.front()<=i-k){
            q.pop_front();
        }
        while(!q.empty()&&a[q.back()]>=a[i]){
            q.pop_back();
        }
        q.push_back(i);
        if(i>=k-1){
            cout<<a[q.front()]<<" ";
        }
    }
    cout<<endl;
    q.clear();
    for(int i=0;i<n;i++){
        if(!q.empty()&&q.front()<=i-k){
            q.pop_front();
        }
        while(!q.empty()&&a[q.back()]<=a[i]){
            q.pop_back();
        }
        q.push_back(i);
        if(i>=k-1){
            cout<<a[q.front()]<<" ";
        }
    }
    return 0;
}