Leetcode239. 滑动窗口最大值——优先队列/双端队列

最新推荐文章于 2024-06-14 08:51:34 发布

Grairain

最新推荐文章于 2024-06-14 08:51:34 发布

阅读量130

点赞数

分类专栏： Leetcode刷题文章标签： leetcode queue

本文链接：https://blog.youkuaiyun.com/weixin_45845896/article/details/115488775

版权

Leetcode刷题专栏收录该内容

25 篇文章

订阅专栏

法一：优先队列

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        priority_queue<pair<int, int>> q;
        for(int i = 0; i < k; ++i){
            q.push({nums[i], i});
        }
        vector<int> res;
        res.push_back(q.top().first);
        for(int i = k; i < nums.size(); i++){
            q.push({nums[i], i});
            while(i - q.top().second >=  k){
                q.pop();
            }
            
            res.push_back(q.top().first);
        }

        return res;
    }
};

法二：双端队列

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        deque<int> q;
        q.push_back(0);

        for(int i = 1; i < k; i++){
            while(!q.empty() && nums[i] >= nums[q.back()]){
                q.pop_back();
            }
            q.push_back(i);
        }

        vector<int> res;
        res.push_back(nums[q.front()]);

        for(int i = k; i < nums.size(); i++){
            while(!q.empty() && nums[i] >= nums[q.back()]){
                q.pop_back();
            }
            q.push_back(i);
            while(i - q.front() >= k){
                q.pop_front();
            }
            res.push_back(nums[q.front()]);
            
        }

        return res;
    }
};