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

链接：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窗口位置[1   3  -1] -3   5   3   6   7  1  [3  -1  -3]  5   3   6   7  1   3 [-1  -3   5]  3   6   7  1   3  -1 [-3   5   3]  6   7  1   3  -1  -3  [5   3   6]  7  1   3  -1  -3   5  [3   6   7]​最小值−1−3−3−333​最大值335567​​

输入格式

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

输出格式

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

输入输出样例

输入 #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%50% 的数据，1≤n≤1051≤n≤105；
对于 100%100% 的数据，1≤k≤n≤1061≤k≤n≤106，ai∈[−231,231)ai​∈[−231,231)。

C++代码如下：

#include<bits/stdc++.h>
using namespace std;

int main(){
    int n,k;
    cin>>n>>k;
    vector<int> a(n+1);
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    deque<int> q;
    for(int i=1;i<=n;i++){
        if(q.size()&&i-k+1>q.front()){
            q.pop_front();
        }
        while(q.size()&&a[i]<=a[q.back()]){
            q.pop_back();
        }
        q.push_back(i);
        if(i>=k){
            cout<<a[q.front()]<<" ";
        }
    }
    q.clear();
    cout<<"\n";
    for(int i=1;i<=n;i++){
        if(q.size()&&i-k+1>q.front()){
            q.pop_front();
        }
        while(q.size()&&a[i]>=a[q.back()]){
            q.pop_back();
        }
        q.push_back(i);
        if(i>=k){
            cout<<a[q.front()]<<" ";
        }
    }
    return 0;
}