洛谷 P1886 滑动窗口

时间:2021-06-29 20:54:44

题目描述

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

例如:

The array is [1 3 -1 -3 5 3 6 7], and k = 3.

洛谷 P1886 滑动窗口

输入输出格式

输入格式:

输入一共有两行,第一行为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<cstdio>
#include<cstring>
#define N 10000000
using namespace std;
int n,k,a[N],q[N],p[N];
void MIN()
{
int head=,tail=;
for(int i=;i<k;i++)
{
while(head<=tail&&q[tail]>=a[i])
tail--;
q[++tail]=a[i];
p[tail]=i;
}
for(int i=k;i<=n;i++)
{
while(head<=tail&&q[tail]>=a[i])
tail--;
q[++tail]=a[i];
p[tail]=i;
while(p[head]<=i-k)
head++;
printf("%d ",q[head]);
if(i==n) printf("\n");
}
}
void MAX()
{
int head=,tail=;
for(int i=;i<k;i++)
{
while(head<=tail&&q[tail]<=a[i])
tail--;
q[++tail]=a[i];
p[tail]=i;
}
for(int i=k;i<=n;i++)
{
while(head<=tail&&q[tail]<=a[i])
tail--;
q[++tail]=a[i];
p[tail]=i;
while(p[head]<=i-k)
head++;
printf("%d ",q[head]);
if(i==n) printf("\n");
}
}
int main()
{
cin>>n>>k;
for(int i=;i<=n;i++)
cin>>a[i];
MIN();
MAX(); return ;
}

思路:单调队列~~