Description
给定一个长度为n(n<=10^6)的数组。有一个大小为k的滑动窗口从数组的最左端移动到最右端。你可以看到窗口中的k个数字。窗口每次向右滑动一个数字的距离。下面是一个例子:数组是 [1 3 -1 -3 5 3 6 7], k = 3。
窗口位置 | 最小值 | 最大值 |
---|---|---|
[1 3 -1] -3 5 3 6 7 | -1 | 3 |
1 [3 -1 -3] 5 3 6 7 | -3 | 3 |
1 3 [-1 -3 5] 3 6 7 | -3 | 5 |
1 3 -1 [-3 5 3] 6 7 | -3 | 5 |
1 3 -1 -3 [5 3 6] 7 | 3 | 6 |
1 3 -1 -3 5 [3 6 7] | 3 | 7 |
你的任务是得到滑动窗口在每个位置时的最大值和最小值。
Input
输入包括两行。
第一行包括n和k,分别表示数组的长度和窗口的大小。
第二行包括n个数字。
Output
输出包括两行。
第一行包括窗口从左至右移动的每个位置的最小值。
第二行包括窗口从左至右移动的每个位置的最大值。
8 3 1 3 -1 -3 5 3 6 7
-
Sample Input
-1 -3 -3 -3 3 3 3 3 5 5 6 7
-
Sample Output
这道题是经典的单调队列模板题。单调队列里的元素应该始终保持递增/递减,用单调队列可以快速地找出一个指定区间内的最小值/最大值。这题因为窗口本身是在移动的,所以应该在每一次移动时把已经在窗口之外的元素出队。
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int a[1000001],n,wide,q[1000001],head,tail;
int main(){
scanf("%d%d",&n,&wide);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
head=1;tail=0;
for(int i=1;i<=n;i++)
{
while(tail>0&&head<=tail&&i-q[head]>=wide)//对于超出窗口范围的元素,出队
head++;
while(tail>0&&head<=tail&&a[q[tail]]>a[i])//对于比新元素更大的元素,出队
tail--;//通过这个操作能始终保持范围内最小的元素位于队首,且队列里的元素是单调递增的
q[++tail]=i;//入队
if(i>=wide) printf("%d ",a[q[head]]);//在元素已经足够多个时输出
}
head=1;tail=0;
memset(q,0,sizeof(q));
printf("\n");
for(int i=1;i<=n;i++)//同上寻找最小值的方法寻找最大值
{
while(tail>0&&head<=tail&&i-q[head]>=wide)
head++;
while(tail>0&&head<=tail&&a[q[tail]]<a[i])
tail--;
q[++tail]=i;
if(i>=wide) printf("%d ",a[q[head]]);
}
return 0;
}
洛谷原题传送门:P1886 滑动窗口 /【模板】单调队列