题目链接:
题目描述:
现在有一堆数字共N个数字(N<=10^6),以及一个大小为k的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
题解:
这道题是典型的单调队列,用STL或数组模拟都可以实现,我这里用的是STL。
所谓单调队列,就是队列里所有的数的值都是单调递增或单调递减的。这道题要求最大值和最小值,显然它们用的方法是一样的,这里我就详细解释一下最小值的求法。
以样例为例:
{1 3 -1 -3 5 3 6 7}
首先从back插入1
队列{1} 编号{1}
插入3,3>1,所以直接从back插入3
队列(1 3) 编号(1 2)
此时应该插入-1,但由于-1<3,3在此后必定不会再成为最小值了,所以将它出队。
同理,1出队
队列{-1} 编号{3}
此时应该插入-3,-3<-1,-1出队
队列{-3} 编号{4}
此时应该插入5,5>-3,所以直接入队
队列{-3 5} 编号{4 5}
此时插入3,3<5,5在此后必定不会再成为最小值了,所以将5出队
队列{-3 3} 编号{4 6}
直接插入6
队列{-3 3 6} 编号{4 6 7}
7>6,此时应该直接插入7,但我们发现如果插入7队列里就有4个数了,队首的-3会不在浮窗的范围内,那么记录它就没有意义了,所以将它出队。
队列{3 6 7} 编号(6 7 8)
由于我们要比较队列的长度和浮窗出度,所以只用开一个记录编号的队列,权值就是a[q.front()]或a[q.bacj()]。
简单来说,整个模拟过程中只有3个操作。
1:a[i]>=a[q.back()]
直接从队尾入队
2:a[i]<a[q.back()]
把队尾出队,然后再入队
3.q.front()+m-1<i
把队首出队,然后再入队
需要注意的是,在2、3操作时,我们还要判断队列是否为空
代码:
#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+!0;
deque<int>maxx,minx;
int n,m,a[maxn];
int amin[maxn],amax[maxn];
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i];
}
maxx.push_front(1);
minx.push_front(1);
amin[1]=1,amax[1]=1;
for(int i=2;i<=n;i++){
while(maxx.size()&&maxx.front()+m-1<i){
maxx.pop_front();
}
while(minx.size()&&minx.front()+m-1<i){
minx.pop_front();
}
while(maxx.size()&&a[maxx.back()]<=a[i]){
maxx.pop_back();
}
while(minx.size()&&a[minx.back()]>=a[i]){
minx.pop_back();
}
maxx.push_back(i);
minx.push_back(i);
amax[i]=maxx.front();
amin[i]=minx.front();
}
for(int i=m;i<=n;i++){
cout<<a[amin[i]]<<" ";
}
cout<<endl;
for(int i=m;i<=n;i++){
cout<<a[amax[i]]<<" ";
}
return 0;
}
写作时间:
2019-8-1