天天看点

239. 滑动窗口最大值239. 滑动窗口最大值

239. 滑动窗口最大值

1.题目描述

给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回滑动窗口中的最大值。

示例:

239. 滑动窗口最大值239. 滑动窗口最大值

2.思路(双向队列)

使用双向队列max_index存储下标。max_index有以下特点:

1.变量的最前端(也就是 max_index.front())是此次遍历的最大值的下标

2.当我们遇到新的数时,将新的数和双项队列的末尾(也就是max_index.back())比较,如果末尾比新数小,则把末尾扔掉,直到该队列的末尾比新数大或者队列为空的时候才停止,做法有点像使用栈进行括号匹配。

3.双项队列中的所有值都要在窗口范围内

特点一只是方便我们获取每次窗口滑动一格之后的最大值,我们可以直接通过 max_index.front() 获得

通过特点二,可以保证队列里的元素是从头到尾降序的,由于队列里只有窗口内的数,所以他们其实就是窗口内第一大,第二大,第三大… 的数。

特点三,因为只要窗口内第一大元素也就是这个 max_index.front() 在窗口内,那我们可以不用管第二大第三大元素在不在区间内了。因为答案一定是这个第一大元素。如果 max_index.front() 不在窗口内,则将其弹出,第二个大元素变成第一大元素,第三大元素变成第二大元素以此类推。

3.代码

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        vector<int> res;
        if(nums.empty()){
            return res;
        }
        deque<int> max_index;
        for(int i = 0;i < nums.size();++i){
            while(!max_index.empty() && nums[i] >= nums[max_index.back()]){//保证双向队列是递减的
                max_index.pop_back();
            }
            max_index.push_back(i);
            if(max_index.front() == (i-k)){
                max_index.pop_front();
            }
            if(i >= k-1){
                res.push_back(nums[max_index.front()]);
            }
        }
        return res;
    }
};
           

4.复杂度分析

时间复杂度:O(n),每个元素被处理两次- 其索引被添加到双向队列中和被双向队列删除。

空间复杂度:O(n),输出数组使用了O(N−k+1) 空间,双向队列使用了O(k)。