滑动窗口——单调队列问题
——HM
现在有一堆数字共N个数字(N<=10^6),以及一个大小为k的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
例如:The array is [1 3 -1 -3 5 3 6 7], and k = 3.
输入输出格式
输入格式:
输入一共有两行,第一行为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
暴力思路:
穷举每一次窗口滑动后的数组,再将数组求最大最小值。估计时间复杂度O(mn),有望拿40分。
正解思路:
这题是一个单调队列的模板题。
首先先讲一下单调队列,说白了就是一个固定单调性的数列,看上去很没用,但单调队列所解决的问题时间复杂度都是线性的,极快,不会超时。
这题怎么体现单调队列呢?很简单,每当窗口滑动时,将新增的那个元素插入队列,为使队列保持单调性,需要将这个元素与队列的后面的元素进行比较,比其小的就将它删去,直到队列是单调性的,这是队首元素便是所求值;当然,窗口移动时还会排除一个元素,这是要对队列进行查封,将那些“又老又不中用”的元素枪毙掉,保持单调队列中的元素的正确性即可。求最大值重复一次上述操作。
注:队列中其实存下标就行。
AC代码:
#include <bits/stdc++.h>
using namespace std;
const int SIZE=1000005;
void workmin();
void workmax();
int read();
void write(int x);
int q2[SIZE],q1[SIZE],a[SIZE],n,m,head,tail;
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++) scanf("%d",&a[i]);
workmin();
workmax();
return 0;
}
void workmin() //求最小值
{
int head=1,tail=0;
for (int i=1;i<=n;i++){
while (head<=tail && q1[head]+m<=i) head++; //枪毙掉太老的元素
while (head<=tail && a[q1[tail]]>a[i]) tail--; //将新入队的元素插入单调队列
q1[++tail]=i; //入队
if (i>=m) printf("%d ",a[q1[head]]); //打印队首元素
}
cout<<endl;
}
void workmax() //求最大值,过程见上
{
int head=1,tail=0;
for (int i=1;i<=n;i++){
while (head<=tail && q2[head]+m<=i) head++;
while (head<=tail && a[q2[tail]]<a[i]) tail--;
q2[++tail]=i;
if (i>=m) printf("%d ",a[q2[head]]);
}
}