题目描述
给定一个大小为$n≤10^6$ 的数组。
有一个大小为 $k$ 的滑动窗口,它从数组的最左边移动到最右边。
你只能在窗口中看到 $k$ 个数字。
每次滑动窗口向右移动一个位置。
以下是一个例子:
该数组为 [1 3 -1 -3 5 3 6 7]
,$k$ 为 $3$。
你的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值。
输入格式
输入包含两行。
第一行包含两个整数 $n$ 和 $k$,分别代表数组长度和滑动窗口的长度。
第二行有 $n$ 个整数,代表数组的具体数值。
同行数据之间用空格隔开。
输出格式
输出包含两个。
第一行输出,从左至右,每个位置滑动窗口中的最小值。
第二行输出,从左至右,每个位置滑动窗口中的最大值。
输入样例:
8 3
1 3 -1 -3 5 3 6 7
输出样例:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
代码&思路
#include<iostream>
using namespace std;
const int N = 1000010;
int q[N],a[N];
int hh,tt=-1,n,k;
int main()
{
scanf("%d%d", &n, &k);//这里建议使用scanf/printf,会比cin/cout快好多!
for (int i = 0; i < n; i ++ )
{
scanf("%d", &a[i]);
}
for (int i = 0; i < n; i ++ )
{
if ( hh <= tt && i - k + 1 > q[hh]) hh ++;
//判断队列是否为空并且判断队列长度是否超过窗口长度
while ( hh <= tt && a[q[tt]] >/*=*/ a[i] ) tt --;
//这里主要判断对位的元素是否大于(等于)当前元素,如果大于(等于),就删除,直到删到小于(等于)当前数为止
//这一步可以理解为:比当前数大的元素留着没什么用,因为求的是最小值,只要留着比较小的就行了
q[++tt] = i;//注意先加入,因为这个数也可能是最小值
if(i >= k - 1) printf("%d ",a[q[hh]]);
//因为是一个一个放入队列的,所以一开始的时候并不是题目的位置,要特判一下
}
printf("\n");
//下面就和上面一样的啦,只不过求的是最大值,过程你们按着上面的反着说吧,我就不写啦~
hh = 0,tt = -1;
for (int i = 0; i < n; i ++ )
{
if ( hh <= tt && i - k + 1 > q[hh]) hh ++;
while ( hh <= tt && a[q[tt]] </*=*/ a[i] ) tt --;
q[++tt] = i;
if(i >= k - 1) printf("%d ",a[q[hh]]);
}
}