单调队列的常见问题,假设有一个窗口大小为3,求其中的最大值,先将数组的前三个数存入队列,找最大值需要o(k),再挪动再找,找完需要o(nk)的复杂度若k是n/2那么复杂度还是会达到o(n2)。想想如何优化,用队列,第一个数进入,进入第二个数的时候如果第二个数比第一个数大的话,第一个数是不是就没有用了,因为后面的最大值也不会是第一个,第一个窗口的最大值也不会是第一个数,所以我们直接将第一个数出队。这样每次移动的时候判断队列中的第一个数就是窗口的最大值。
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int>ans;
deque<int>q;//用双端队列存储进入队列的数组下标
for(int i=0;i<nums.size();i++)
{
while(!q.empty()&&nums[i]>=nums[q.back()])q.pop_back();//入队的数如果更大,就将比他小的出队
q.push_back(i);//将下标入队
if(i-q.front()>=k)q.pop_front();//当前队头不在窗口中就需要从前面出队
if(i>=k-1)ans.push_back(nums[q.front()]);
}
return ans;
}
};