239. 滑动窗口最大值 --19/11/13

最新推荐文章于 2024-08-20 23:30:00 发布

敲代码的拉布拉多

最新推荐文章于 2024-08-20 23:30:00 发布

阅读量142

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_44357371/article/details/103047429

版权

LeetCode 专栏收录该内容

16 篇文章 1 订阅

订阅专栏

题目：
在这里插入图片描述

方法一：

利用大顶堆，每次把k个数放进堆里，往右移，就把第一个数移出，后面的数加入，数组里面加入堆顶元素。

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        PriorityQueue<Integer> maxHeap = new PriorityQueue<Integer>(new Comparator<Integer>() {
            public int compare(Integer i1,Integer i2){
                return i2-i1;
            }
        });
        //初始化堆
        if(nums.length==0){
            int[] arr = new int[0];
            return arr;
        }
        if(maxHeap.size()==0){
            for(int i=0;i<k;i++){
                maxHeap.add(nums[i]);
            }
        }
        int[] array = new int[nums.length-k+1];
        for(int i=0;i<nums.length-k;i++){
            array[i] = maxHeap.peek();
            maxHeap.remove(nums[i]);
            maxHeap.add(nums[i+k]);
        }
        if(maxHeap.size()!=0){
            array[nums.length-k] = maxHeap.peek();
        }
        return array;
    }
}

在这里插入图片描述

方法二：

利用双向队列，双向队列存储最大元素的下标，每一次在滑动窗体中，如果新加元素比前面的元素大，就把前面元素清除，最终保证最左边元素是最大的

class Solution {
    ArrayDeque<Integer> deq = new ArrayDeque<Integer>();
    int[] nums;
    public int[] maxSlidingWindow(int[] nums, int k) {
        if(nums.length*k == 0)  return new int[0];
        if(k==1) return nums;

        this.nums = nums;
        int max_index=0;
        for(int i =0;i<k;i++){
            clean(i,k);
            deq.addLast(i);
            if(nums[i]>nums[max_index]){
                max_index = i;
            }
        }
        int[] output = new int[nums.length-k+1];
        output[0] = nums[max_index];
        for(int i=k;i<nums.length;i++){
            clean(i,k);
            deq.addLast(i);
            output[i-k+1] = nums[deq.getFirst()];
        }
    return output;
    }
    public void clean(int i,int k){
        if (!deq.isEmpty() && deq.getFirst() == i - k)
            deq.removeFirst();
        while(!deq.isEmpty() && nums[i] > nums[deq.getLast()]){
            deq.removeLast();
        }
    }
}