luogu P1886 滑动窗口

题目描述

现在有一堆数字共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

单调队列luo

#include <iostream>
#include <cstdio>

using namespace std;
const int N = 1e6 + 10;

int a[N],q[N];
int n, k, head, tail;

inline int read()
{
    int x = 0, f = 1; char c = getchar();
    while(c < '0' || c > '9'){if(c == '-')f = -1;c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}

int main()
{
    n = read();
    k = read();
    for(int i= 1; i<= n; i ++)
        a[i] = read();
    head = 1, tail = 0;
    for(int i = 1; i <= n; i ++)
    {
        while(head <= tail && q[head] < i - k + 1)
            head ++;
        while(head <= tail && a[q[tail]] > a[i])
            tail --;
        q[++ tail] = i;
        if(i >= k)
            printf("%d ",a[q[head]]);
    }
    printf("
");
    head = 1, tail = 0;
    for(int i = 1; i <= n; i ++)
    {
        while(head <= tail && q[head] < i - k + 1)
            head ++;
        while(head <= tail && a[q[tail]] < a[i])
            tail --;
        q[++ tail] = i;
        if(i >= k)
            printf("%d ",a[q[head]]);
    }
    return 0;
}