要求:
给定一组数和滑动窗口的大小,
从左至右,输出每个位置滑动窗口中的最小值,
从左至右,输出每个位置滑动窗口中的最大值。
样例:
输入:
8 3
1 3 -1 -3 5 3 6 7
输出:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
- 队列中存放的是数组下标
- 队列前面的值对应数组中的数一定小于队列后面的值对应数组中的数(队头元素对应的数组中的数就是当前窗口中的最小值
- 最小值和最大值的代码只需修改>=为<=即可达到队列单调效果
- 必须先将当前下标存入队列再取队头元素,因为当前元素可能就是窗口的最小值
- 单调队列题较少
#include<iostream>
#include<cstdio>
using namespace std;
const int N=1000010; //数据范围是10的6次方
int n,k; // n个数,窗口长度为k
int a[N],q[N]; // 数组a存放输入的数据,数组q表示队列(存放下标)
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 && q[hh]<i-k+1) hh++; //队头超出窗口,删除队头
while(hh<=tt && a[q[tt]]>=a[i]) tt--; // 队尾元素对应的值不小于当前数,删除
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 && q[hh]<i-k+1) hh++;
while(hh<=tt && a[q[tt]]<=a[i]) tt--;
q[++tt]=i;
if(i>=k-1) printf("%d ",a[q[hh]]);
}
puts("");
return 0;
}
欢迎指正错误^0^