0
点赞
收藏
分享

微信扫一扫

剑指 Offer 59 - I. 滑动窗口的最大值

软件共享软件 2022-03-12 阅读 31

如果我们用一个数组加一个列表会超出内存限制的!!!所以我们可以用数组加双端队列来解答。

首先我们需要确保双端队列要是单调递减的,所以队列的首个元素就是当前窗口的最大值。

我们将第一个窗口的所有值,按照单调队列的规则入队之后,因为队列为单调递减,所以队头元素必为当前窗口的最大值,则将队头元素添加到数组中。即判断队列的尾元素是否大于窗口值,如果否那么在队列中弹出。

然后移动窗口,要判断当前窗口前的元素是否和队列首元素相等。如果相等那么就将这个值从双端队列中弹出。

每次将队头元素加入到数组中去。

最后返回数组。

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        int len = nums.length;
        if (len == 0) {
            return nums;
        }
        int[] arr = new int[len - k + 1];
        int arr_index = 0;
        //我们需要维护一个单调递增的双向队列
        Deque<Integer> deque = new LinkedList<>();
        //先将第一个窗口的值按照规则入队
        for (int i = 0; i < k; i++) {
             while (!deque.isEmpty() && deque.peekLast() < nums[i]) {
               deque.removeLast();
            }
            deque.offerLast(nums[i]);
        }
        //存到数组里,队头元素
        arr[arr_index++] = deque.peekFirst();
        //移动窗口
        for (int j = k; j < len; j++) {
            //对应咱们的红色情况,则是窗口的前一个元素等于队头元素
            if (nums[j - k] == deque.peekFirst()) {
                deque.removeFirst();
            }
            while (!deque.isEmpty() && deque.peekLast() < nums[j]) {
                deque.removeLast();
            }
            deque.offerLast(nums[j]);
            arr[arr_index++] = deque.peekFirst();
        }
        return arr;
    }
}
举报

相关推荐

0 条评论