P1886 滑动窗口单调队列

佚名 6年前 (2019-04-17) 随笔 851人围观抢沙发百度已收录

题目描述

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

例如：

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。

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 <iostream>
#include <queue>
using namespace std;
const int MAXN = 1e6 + 8;
int a[MAXN];
struct node{
    int t;
    int v;
};
deque<node> q1,q2;
int main(){
    ios::sync_with_stdio(false);
    int n,k;
    int tmp;
    cin >> n >> k;
    for(int i=1;i<=n;++i){
        cin >> tmp;
        a[i] = tmp;
    }
    for(int i=1;i<=n;i++){
        //min
        while(q1.size() && q1.back().v >= a[i])
            q1.pop_back();
        q1.push_back(node{i,a[i]});
        if(i - q1.front().t >= k) 
            q1.pop_front();
        if(i>=k) 
            cout << q1.front().v << (i!=n?" ":"");
    }
    cout << endl;
    for(int i=1;i<=n;i++){
        //max
        while(q2.size() && q2.back().v <= a[i])
            q2.pop_back();
        q2.push_back(node{i,a[i]});
        if(i - q2.front().t >= k) 
            q2.pop_front();
        if(i>=k) 
            cout << q2.front().v << (i!=n?" ":"");        
    }
    return 0;
}