队列的两个基本操作:入队 将一个数据放到队列尾部;出队 从队列的头部取出一个元素。
题解:
https://www.acwing.com/solution/acwing/content/853/
首先,最直接的做法当然是模拟滑动窗口的过程,每向右滑动一次都遍历一遍窗口内的数字找最大的输出,这样的复杂度是O(kn)
,我们可以考虑优化一下。窗口向右滑动的过程实际上就是将处于窗口的第一个数字删除,同时在窗口的末尾添加一个新的数字,这就可以用双向队列来模拟,每次把尾部的数字弹出,再把新的数字压入到头部,然后找队列中最大的元素即可。
为了更快地找到最大的元素,我们可以在队列中只保留那些可能成为窗口最大元素的数字,去掉那些不可能成为窗口中最大元素的数字。考虑这样一个情况,如果队列中进来一个较大的数字,那么队列中比这个数更小的数字就不可能再成为窗口中最大的元素了,因为这个大的数字是后进来的,一定会比之前早进入窗口的小的数字要晚离开窗口,那么那些早进入且比较小的数字就“永无出头之日”,所以就可以弹出队列。
于是我们维护一个双向单调队列,队列放的是元素的下标。我们假设该双端队列的队头是整个队列的最大元素所在下标,至队尾下标代表的元素值依次降低。初始时单调队列为空。随着对数组的遍历过程中,每次插入元素前,首先需要看队头是否还能留在队列中,如果队头下标距离i超过了k,则应该出队。其次需要维护队列的单调性,如果nums[i]大于或等于队尾元素下标所对应的值,则当前队尾再也不可能/没有机会充当某个滑动窗口的最大值了,故需要队尾出队。始终保持队中元素从队头到队尾单调递减。依次遍历一遍数组,每次队头就是每个滑动窗口的最大值所在下标。
时间复杂度分析:每个元素最多入队出队一次,复杂度为O(n)
class Solution { public: vector<int> maxInWindows(vector<int>& nums, int k) { vector<int>res;//保存答案的数组 deque<int>q; for(int i = 0; i < nums.size(); i++){ if(!q.empty() && i-q.front() >= k)//判断队头是否需要出队,需要出栈,执行pop_front() q.pop_front();
//维护队列单调性,如果队尾元素nums[q.back()]比要添加的元素小,一直删除队尾元素,直至不满足while while(!q.empty()&&nums[q.back()]<nums[i]) q.pop_back();
//将要添加的元素的下标添加进队列q队尾。(上一个while维护了队列单调性,保证执行这一句时,插入队尾的元素是队列中的最小值,因为比nums[i]小的都在上一个while中被删除了) q.push_back(i); if(i >= k-1){//i >= k-1:表示遍历数组长度大于滑动窗口长度 res.push_back(nums[q.front()]);//取队头作为窗口最大元素 } } return res; } }; 作者:cornerCao 链接:https://www.acwing.com/solution/acwing/content/853/ 来源:AcWing 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。