滑动窗口最大值
问题重述:
给定一个大小为n≤106的数组。
有一个大小为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
思路
暴力算法时间复杂度为O(nk)容易超时,可以采用单调队列的方法来解决,我们可以定义两个双端队列,分别处理最小值和最大值;
核心代码一共分四步:
第一步:每次读入一个数时判断队尾的数据是否一定不满足要求(如若找最大值判断储存最大值的双端队列队尾元素是不是小于等于当前数,如果是那么它就一定不满足要求),当她一定不满足要求是我们就将它弹出,直至找到一个满足要求的或将队列弹空为止
第二步:将当前数的下标存入队列
第三步:判断队头元素是不是不在窗口内了(和当前数下标差大于等于k),若不在了则将队头弹出,直至找到一个窗口内的为止(队列不会被弹空因为当前数已经入队了,它必在自己所在的窗口内,所以这个数一定会被找到)
第四步:当i>=k-1时(即已形成窗口)记录当前窗口的最值答案
代码:
#include<iostream>
#include<deque>
#include<vector>
using namespace std;
const int N=1e5+10;
int n,k,nums[N];
deque<int>maxque,minque;//两个双端队列做窗口分别处理最大值最小值
vector<int>minans,maxans;//两个答案数组
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cin>>n>>k;
for(int i=0;i<n;i++)
{
cin>>nums[i];//输入数据
//第一步:判断队尾是不是一定不符合要求,若是则弹出
while(!maxque.empty()&&nums[maxque.back()]<=nums[i])maxque.pop_back();
while(!minque.empty()&&nums[minque.back()]>=nums[i])minque.pop_back();
//第二步:将当前数的下标入队
maxque.push_back(i);
minque.push_back(i);
//第三步:判断队头是否已经不在窗口内,若是则弹出
if(i-maxque.front()>=k)maxque.pop_front();
if(i-minque.front()>=k)minque.pop_front();
//第四步:若当前已经形成窗口记录最值答案
if(i>=k-1)
{
minans.push_back(minque.front());
maxans.push_back(maxque.front());
}
}
//输出结果
for(auto a:minans)cout<<nums[a]<<" ";
cout<<endl;
for(auto a:maxans)cout<<nums[a]<<" ";
return 0;
}