P1886 滑动窗口（单调队列）

滑动窗口

传送门

题目描述

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

例如：

The array is [1,3,-1,-3,5,3,6,7], and k = 3。

输入格式

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

输出格式

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

输入样例

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

输出样例

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

【数据范围】

对于 50% 的数据，$1\le n\le 10^5$；
对于 100% 的数据，$1\le k\le n\le 10^6$，$a_i\in [-2^{31},2^{31})$。

思路

用deque来维护一个递增的序列和一个递减的序列，每次入队时清理队头、队尾的无效元素。剩下的队头元素就是最大值、最小值了。

#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=1e6+5;
struct N{int v,p;};
deque<N> big,sma;
int n,k,v,ansb[maxn],anss[maxn];

int main(){
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++){
        scanf("%d",&v);
        while(big.size()&&big.back().v>=v)   big.pop_back();
        while(big.size()&&i-big.front().p>=k) big.pop_front();
        big.push_back({v,i});
        while(sma.size()&&sma.back().v<=v)   sma.pop_back();
        while(sma.size()&&i-sma.front().p>=k) sma.pop_front();
        sma.push_back({v,i});
        ansb[i]=sma.front().v;
        anss[i]=big.front().v;
    }
    for(int i=k-1;i<n;i++)  printf("%d%c",anss[i],i==n-1?'\n':' ');
    for(int i=k-1;i<n;i++)  printf("%d%c",ansb[i],i==n-1?'\n':' ');
}