Problem
Solution
题意为给定一个数组,给定一个长度为k滑动窗口,这个滑动窗口从数组最左边滑到数组最右边,每次滑动的时候,让你求位于滑动窗口内的数字的最大值。
此题最暴力的解法就是,枚举所有起点i,每次都从i开始遍历k个数字找出最大值,时间复杂度o(n^2)
如果利用单调队列,那么我们就可以把时间复杂度降到o(n)
单调队列的队首元素一定是当前滑动窗口内的最大值,整个单调队列是非递减序的,而且由于是顺序扫描的关系,整个单调队列的元素在原数组的下标位置一定是递减的。
为了维护单调队列始终保持以上性质,我们对顺序遍历中碰到的每个元素num[i]做一下操作
1 判断当前的i和队列里原数组下标最小的元素(即队首元素)相差是否超过窗口长度k。用通俗的话来说,就是滑动窗口向右滑动了,窗口长度固定为k,那么你最左边原来位于窗口内现在位于窗口外的元素就要出队了。
2 当前的元素num[i],要与本来已经在队列里的元素做对比,看队列中的元素与num[i]相比,还是否「有潜力成为」最大值。如果当前元素比队列中的某些元素大,那么这些元素都需要出队,因为在这些元素因1中所述原因出队之前,当前num[i]一定存在于窗口中,这些元素比num[i]小,自然没有成为最大值的潜力了
因为当前元素在窗口最右边的原因,不管元素大小,我们都要把当前的元素入队。
然后判断一下窗口是否有k个元素长了,若有了,则此时队列中的队首元素就是当前窗口的最大值,存入最后的答案之中。
class Solution {
public:
struct num{
int data;
int index;
};
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> ans;
int n=nums.size();
deque<num> ma;
for(int i=0;i<n;++i){
while(!ma.empty() && i-ma.front().index>=k) ma.pop_front();
while(!ma.empty() && ma.back().data<nums[i] ) ma.pop_back();
ma.push_back({nums[i],i});
if(i+1>=k) ans.push_back(ma.front().data);
}
return ans;
}
};