luogu P1886 滑动窗口（单调队列

Posted 2020-10-10 Nico&11101001

题目描述

现在有一堆数字共N个数字（N<=10^6），以及一个大小为k的窗口。现在这个从左边开始向右滑动，每次滑动一个单位，求出每次滑动后窗口中的最大值和最小值。

例如：

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

输入输出格式

输入格式：

 

输入一共有两行，第一行为n,k。

第二行为n个数(<INT_MAX).

 

输出格式：

 

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

第二行为每次窗口滑动的最大值

 

输入输出样例

输入样例#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%的数据，n<=10^5

100%的数据，n<=10^6

单调队列版，单调队列内记录的为下标

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1000003;
int n,k;
int a[maxn];
int q1[maxn],q2[maxn];
int mx[maxn],mn[maxn];
int main() {
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;++i) scanf("%d",a+i);
    int h1,h2,t1,t2;
    q1[1]=q2[1]=1;
    h1=h2=t1=t2=1;int cnt=0;//5 1 1 2 3 4 5
    if(k==1) {
        for(int i=1;i<=n;i++)cout<<a[i]<<" ";
        cout<<endl;
        for(int i=1;i<=n;i++)cout<<a[i]<<" ";return 0;
    }
    for(int i=2;i<=k;++i) {
        while(h1<=t1&&a[q1[t1]]<a[i])t1--;
        while(h2<=t2&&a[q2[t2]]>a[i])t2--;
        q1[++t1]=q2[++t2]=i;
    }
    mx[++cnt]=a[q1[h1]],mn[cnt]=a[q2[h2]];
    for(int i=k+1;i<=n;++i) {
        while(q1[h1]<i-k+1)
        h1++;
        while(q2[h2]<i-k+1)
        h2++;
        while(h1<=t1&&a[q1[t1]]<a[i])t1--;
        while(h2<=t2&&a[q2[t2]]>a[i])t2--;
        q1[++t1]=q2[++t2]=i;
        mx[++cnt]=a[q1[h1]];mn[cnt]=a[q2[h2]];
    }
    for(int i=1;i<=cnt;++i) 
        printf("%d ",mn[i]);
    puts("");
    for(int i=1;i<=cnt;++i)
        printf("%d ",mx[i]);
    return 0;
}