洛谷 P1886 滑动窗口

题目：滑动窗口

思路：用两个单调队列维护。

代码：

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

#define maxn 1000000

int n,m;
int a[maxn+5]={0};

int q1[maxn+5],q2[maxn+5];
int h1=0,t1=0;
int h2=0,t2=0;
vector<int> ans1,ans2;

int main() {
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<m;i++) {
        while(a[q1[t1]]>a[i]&&t1>=h1) t1--;
        q1[++t1]=i;
    }
    a[0]=(1<<30);
    for(int i=1;i<m;i++) {
        while(a[q2[t2]]<a[i]&&t2>=h2) t2--;
        q2[++t2]=i;
    }
    for(int i=m;i<=n;i++) {
        while(a[q1[t1]]>a[i]&&t1>=h1) t1--;
        q1[++t1]=i;
        while(i-q1[h1]>=m) h1++;
        ans1.push_back(q1[h1]);
        while(a[q2[t2]]<a[i]&&t2>=h2) t2--;
        q2[++t2]=i;
        while(i-q2[h2]>=m) h2++;
        ans2.push_back(q2[h2]);
    }
    for(int i=0;i<ans1.size();i++) printf("%d ",a[ans1[i]]);
    printf("\n");
    for(int i=0;i<ans2.size();i++) printf("%d ",a[ans2[i]]);
    return 0;
}