洛谷——P1886 滑动窗口|| POJ——T2823 Sliding Window

https://www.luogu.org/problem/show?pid=1886#sub

||

http://poj.org/problem?id=2823

题目描述

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

using namespace std;

const int N=1234567;
int n,k,a[N];
int num[N],que_min[N],que_max[N];

void read(int &x)
{
    int ch=getchar(),if_=0; x=0;
    while(ch<'0'||ch>'9')
    {
        if(ch=='-') if_=1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    if(if_) x=(~x)+1;
}

void Get_min()
{
    int head=1,tail=0;
    for(int i=1;i<k;i++)
    {
        while(head<=tail&&que_min[tail]>=a[i]) tail--;
        que_min[++tail]=a[i];
        num[tail]=i;
    }
    for(int i=k;i<=n;i++)
    {
        while(head<=tail&&que_min[tail]>=a[i]) tail--;
        que_min[++tail]=a[i];
        num[tail]=i;
        while(num[head]<=i-k) head++;
        printf("%d ",que_min[head]);
    }
}

void Get_max()
{
    int head=1,tail=0;
    for(int i=1;i<k;i++)
    {
        while(head<=tail&&que_max[tail]<=a[i]) tail--;
        que_max[++tail]=a[i];
        num[tail]=i;
    }
    for(int i=k;i<=n;i++)
    {
        while(head<=tail&&que_max[tail]<=a[i]) tail--;
        que_max[++tail]=a[i];
        num[tail]=i;
        while(num[head]<=i-k) head++;
        printf("%d ",que_max[head]);
    }
}

int main()
{
    read(n); read(k);
    for(int i=1;i<=n;i++) read(a[i]);
    Get_min();
    printf("\n");
    Get_max();
    return 0;
}