[单调队列]滑动窗口 /【模板】单调队列
滑 动 窗 口 / 【 模 板 】 单 调 队 列 滑动窗口 /【模板】单调队列 滑动窗口/【模板】单调队列
题目描述
有一个长为 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≤n≤105;
对于 100% 的数据,1≤k≤n≤1061,ai∈[−231,231)。
code
#include<stdio.h>
#include<iostream>
using namespace std;
long long n,k,et,er,h1,h2,tail1,tail2,a[1000005],b[1000005],c[1000005],q1[1000005],q2[1000005],p1[1000005],p2[1000005];
int main(){
scanf("%lld%lld",&n,&k);
for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
q1[1]=q2[1]=a[1];p1[1]=p2[1]=h1=h2=tail1=tail2=1;
if(k==1)b[++et]=c[++er]=a[1];
for(int i=2;i<=n;++i){
while(a[i]<=q1[tail1] and h1<=tail1)tail1--;
q1[++tail1]=a[i];p1[tail1]=i;
while(i-p1[h1]>=k)h1++;
if(i>=k)b[++et]=q1[h1];
while(q2[tail2]<=a[i] and h2<=tail2)tail2--;
q2[++tail2]=a[i];p2[tail2]=i;
while(i-p2[h2]>=k)h2++;
if(i>=k)c[++er]=q2[h2];
}
for(int i=1;i<=et;++i)printf("%lld ",b[i]);
printf("\n");
for(int i=1;i<=er;++i)printf("%lld ",c[i]);
return 0;
}