刷怪日记-俩种方式解决滑动窗口最大值

题目

解题思路

使用大顶堆解决

• 大顶堆是最大值永远处于最大处，而此题就是获取窗口内的最大值
• 先添加k个元素建立堆，再按照上述想法，一直遍历元素然后加入堆中即可，但是此时堆上最大元素并不在滑动窗口内
• 那么在添加元素进堆时，循环判断此时元素是否在滑动窗口内，如果不在则弹出该元素。
• 知道上述想法以后，如何判断元素是否在滑动窗口内，只能依据元素索引来判断，因此我们需要建立元素值和索引的映射关系，eg: [元素值, 索引]，并该关系作为一个元素建立堆

public int[] maxSlidingWindow(int[] nums, int k) {

       i // 堆实现
        int n = nums.length;
        // 大顶堆
        PriorityQueue<int[]> pq = new PriorityQueue<int[]>(new Comparator<int[]>() {
            // 判断俩个元素是否相等，相等则返回true新元素置堆顶，不相等则判断俩者元素大小，新元素大则true，否则false
            public int compare(int[] pair1, int[] pair2) {
                return pair1[0] != pair2[0] ? pair2[0] - pair1[0] : pair2[1] - pair1[1];
            }
        });
        // 堆添加k个元素
        for (int i = 0; i < k; ++i) {
            pq.offer(new int[]{nums[i], i});
        }
        int[] ans = new int[n - k + 1];
        // 滑动窗口第一次满元素，ans数组添加堆顶元素
        ans[0] = pq.peek()[0];
        for (int i = k; i < n; ++i) {
            pq.offer(new int[]{nums[i], i});
            // 判断该堆顶元素是否在滑动窗口内，不在则弹出
            while (pq.peek()[1] <= i - k) {
                pq.poll();
            }
            // ans数组按位添加元素
            ans[i - k + 1] = pq.peek()[0];
        }
        return ans;
                                              
    }

单调队列

• 此题主要使用单调队列(双端队列)解决，头出尾进尾出
• 判断队列元素是否在滑动窗口内
• 保持单调队列单调递减，从大到小， 进来的元素一定要比队尾元素大
• 初始结果组，滑动窗口先要满足k的大小在添加元素到结果组

public int[] maxSlidingWindow(int[] nums, int k) {
ArrayDeque<Integer> deque = new ArrayDeque<>();
        int n = nums.length;
        int[] res = new int[n - k + 1];
        int idx = 0;
        for(int i = 0; i < n; i++) {
            // 根据题意，i为nums下标，是要在[i - k + 1, i] 中选到最大值，只需要保证两点
            // 1.队列头结点需要在[i - k + 1, i]范围内，不符合则要弹出
            while(!deque.isEmpty() && deque.peek() < i - k + 1){
                deque.poll();
            }
            // 2.既然是单调，就要保证每次放进去的数字要比末尾的都大，否则也弹出
            while(!deque.isEmpty() && nums[deque.peekLast()] < nums[i]) {
                deque.pollLast();
            }

            deque.offer(i);

            // 因为单调，当i增长到符合第一个k范围的时候，每滑动一步都将队列头节点放入结果就行了
            if(i >= k - 1){
                res[idx++] = nums[deque.peek()];
            }
        }
        return res;
        }