P1886 滑动窗口（单调队列）

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

code

#include<cstdio>
#include<cstring>

int q[1000100];
int t[1000100];
int v[1000100];
int n,m,head,tail;
void work_min()
{
    head = 1;
    tail = 0;
    for (int i=1; i<=n; ++i)
    {
        while (head<=tail&&q[tail]>=t[i]) tail--;
        q[++tail] = t[i];
        v[tail] = i;
        while (v[head]<=i-m) head++;
        if (i>=m) printf("%d ",q[head]);
    }
    printf("
");
}
void work_max()
{
    memset(q,0,sizeof(q));
    memset(v,0,sizeof(v));
    head = 1;
    tail = 0;
    for (int i=1; i<=n; ++i)
    {
        while (head<=tail&&q[tail]<=t[i]) tail--;
        q[++tail] = t[i];
        v[tail] = i;
        while (v[head]<=i-m) head++;
        if (i>=m) printf("%d ",q[head]);
    }
    printf("
");
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1 ;i<=n; ++i)
        scanf("%d",&t[i]);
    work_min();
    work_max(); 
    return 0;
}