LeetCode 239: 滑动窗口最大值(困难)

LeetCode 239: 滑动窗口最大值

在这里插入图片描述
在这里插入图片描述

解答

代码一:大根堆

最好的解法是单调队列,但是一开始想的是大根堆

问题在于如何将滑动窗口移出的元素从大根堆中删除,大根堆只能pop,没法移除特定的元素。这里就陷入了困境。

解决方法是判断top是否在滑动窗口内,因此要将下标放入大根堆中,判断top->second是否大于i-k

滑动窗口的大小永远小于等于k,但是这里大根堆的大小最坏情况是O(n),自己没有想到上面的转换,

很大程度上我就是既想删除移出的元素,又想保持大根堆元素小于等于k,陷入了困境。现在是pop堆顶并判断是否在滑动窗口内,而且空间复杂度O(n)

// 作者:LeetCode-Solution
class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        int n = nums.size();
        priority_queue<pair<int, int>> q;
        for (int i = 0; i < k; ++i) {
            q.emplace(nums[i], i);
        }
        vector<int> ans = {q.top().first};
        for (int i = k; i < n; ++i) {
            q.emplace(nums[i], i);
            while (q.top().second <= i - k) {
                q.pop();
            }
            ans.push_back(q.top().first);
        }
        return ans;
    }
};

代码二:单调队列

单调队列典型题目,
单调队列维护一个区间内元素单调递增/递减,在元素入队的时候比较像单调栈,在元素移出滑动窗口时可以从另一端pop
里面有一个细节是单调队列中存储的是下标索引,这样可以判断front是否处于滑动窗口之外

deque平时用的少,这里记录一下常用方法
在这里插入图片描述

// my solution
class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        if(k==1) return nums;
        vector<int> ans(nums.size()-k+1, -1);
        deque<int> dq;

        // [0, k-1)预处理
        int now = 0, idx=0;
        for(int i=0; i<k-1; i++){
            now = nums[i];
            if(dq.empty()) dq.push_back(i);
            else if(now < nums[dq.back()]) dq.push_back(i);
            else{ // now >= dp.back()
                while(!dq.empty() && now>= nums[dq.back()]){
                    dq.pop_back();
                }
                dq.push_back(i);
            }
        }

        for(int i=k-1; i<nums.size(); i++){
            now = nums[i];
            // windows scope: [i-k+1,i]
            if(dq.front() < i-k+1) dq.pop_front();

            if(dq.empty()) dq.push_back(i);
            else if(now < nums[dq.back()])  dq.push_back(i);
            else{ // now >= dp.back()
                while(!dq.empty() && now>=nums[dq.back()])
                    dq.pop_back();
                dq.push_back(i);
            }
            ans[idx++] = dq.front();
        }
        
        for(int i=0; i<ans.size(); i++)
            ans[i] = nums[ans[i]];

        return ans;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值