题目描述
有一个大小为k的滑动窗口,它从数组的最左边移动到最右边。
您的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值
输入格式
第一行包含两个整数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
- 先看暴力如何求,即循环遍历所有元素,这样算法时间复杂度为O(nk)
- 去掉没有用到的元素,观察剩下的元素
- 剩下的元素是单调的,用单调队列来做
规律
1 3 -1 -3 5 3 6 7
输出样例:
-1 -3 -3 -3 3 3
可以看出1 3(第一个) 5 6 7没有用到
也就是当前面元素比要入队列的元素大的时候,就把前面的元素删除
这样队头就一直是最小的元素值
当满足k个的时候输出
数组模拟队列模板
int q[N];
//hh, tt 分别表示队头和队尾
int hh = 0, tt = -1;
//队尾插入元素
q[++ tt] = x;
//队头删除元素
hh ++;
//队头的值
q[hh];
代码
#include<iostream>
using namespace std;
const int N = 100010;
int a[N], q[N];
int n, k;
int main(){
scanf("%d%d", &n, &k);
for(int i = 0; i < n; i ++ ) scanf("%d", &a[i]);
int 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 --;// 循环到的数组的值a[i]比队尾元素小时,就让弹出队尾元素
q[++ tt] = i;
if(i >= k - 1) printf("%d ", a[q[hh]]);
}
puts("");
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]]);
}
return 0;
}
菜b