代码随想录算法训练营第13天 | LeetCode239.滑动窗口最大值

题目链接:239. 滑动窗口最大值 - 力扣(LeetCode)

作者思考:

这种题目一拿到手就应该想到暴力算法行不通,因为滑动窗口的时间复杂度就是O(n)了,而且要比较出滑动窗口内的最大值,就和滑动窗口的大小k有关,那暴力算法的时间复杂度就是O(kn),且k的取值可以取到很大,这样暴力算法一定会超时。

那么我们需要一个队列,这个队列呢,放进窗口里的元素,然后随着窗口的移动,队列也一进一出,每次移动后,队列告诉我们里面的最大值是什么。那这个队列应该长这样

class MyQueue{
    void add(int value) {
    }
    void poll(int value) {
    }
    int peek() {
        return myQueue.peek();
    }
}

每次窗口移动的时候,调用myQueue.poll(滑动窗口中移除元素的数值),myQueue.add(滑动窗口中添加元素的数值),然后myQueue.peek()直接返回当前队列中的最大值。这题题不就解决了。收工下班!!!那我们现在要找,有没有这种现成的数据结构可以拿过来直接用呢?可惜并没有,也就是说,我们需要自己实现这个队列并完成上述操作。

再分析一下这个队列我们需要给他进行大概的排序,也就是说队列的头部一定是该队列中的最大值,那我们怎么样才能将插入队列的元素中的最大值排到队列头部呢?(可以创建一个双端队列,队列的两端都可以进行出队和入队的操作)

难点:如何才能保证目前队列中的最大值在队列的头部

 根据这个动画可以很形象的看出,当我们插入元素时,如果插入元素(val)大于当前队列中的元素(deque.peek())将之前的元素全部poll出去,也就是说val是队列现在的最大值,且在当前队列的出口;再加入的元素如果没有比val大,就正常插入到val后。(这就是要维护队列中最大元素)

 

class MyQueue{
    Deque<Integer> deque = new LinkedList<>();//双端队列 两端都可以进行入队出队
    void poll(int val) {
        //判断当前要弹出的元素的数值是否等于队列出口的数值,如果相等则弹出
        //同时判断队列是否为空
        if (!deque.isEmpty() && val == deque.peek()) {
            deque.poll();
        }
    }

    //添加元素时,val大于队列中的最后一个元素,就弹出
    //保证队的出口处是最大值
    void add(int val) {
        while (!deque.isEmpty() && val > deque.getLast()) {
            deque.removeLast();
        }
        deque.add(val);
    }

    int peek() {
        return deque.peek();
    }
}

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        if (nums.length == 1) {
            return nums;
        }
        int len = nums.length - k +1;//存入结果数组的长度
        int[] res = new int[len];
        MyQueue myQueue = new MyQueue();
        int count = 0;
        for (int i = 0; i < k; i++) {
            myQueue.add(nums[i]);//将数组元素添加到队列中
        }
        res[count++] = myQueue.peek();//取出队顶的值(该队列的最大值)
        for (int i = k; i < nums.length; i++) {
            myQueue.poll(nums[i - k]);
            myQueue.add(nums[i]);
            //记录当前最大值
            res[count++] = myQueue.peek();
        }
        return res;
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值