剑指 Offer 59 - I. 滑动窗口的最大值(复习)

题目:

给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。

示例:

输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7]
解释:

滑动窗口的位置 最大值
--------------- -----
[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7

 

解答:

 

 

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        if(nums.length<k || k<=0) return new int[0];
        Deque<Integer> deque = new LinkedList<>();

        int[] res = new int[nums.length-k+1];
        for(int i = 0;i<k;i++){
            while(!deque.isEmpty() && deque.peekLast()<nums[i])
                deque.removeLast();
            deque.addLast(nums[i]);
        }

        res[0] = deque.peekFirst();
        for(int j = k;j<nums.length;j++){
            if(deque.peekFirst() == nums[j-k])
                deque.removeFirst();
            while(!deque.isEmpty() && deque.peekLast()<nums[j])
                deque.removeLast();
            deque.addLast(nums[j]);

            res[j-k+1] = deque.peekFirst();
        }

        return res;
    }
}

 方法二:大根堆 + 双指针

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        if(nums.length<k || k<=0 ||nums.length<=0||nums == null) return new int[0];
        PriorityQueue<Integer> heap = new PriorityQueue<>((o1,o2)->o2-o1);
        int l = 0, r = k-1,index = 0,cnt = 0;
        int[] res = new int[nums.length-k+1];

        for(int j = 0;j<k;j++)
            heap.offer(nums[j]);

        for(;r<nums.length;){
            res[index] = heap.peek();
            index++;
            heap.remove(nums[l]);
            l++;
            r++;
            if(r<nums.length)
                heap.add(nums[r]);
        }

        // for(;cnt<k;cnt++)
        //     heap.offer(nums[cnt]);
        // while(cnt <nums.length){
        //     res[index] = heap.peek();
        //     index++;
        //     heap.remove(nums[cnt-k]);
        //     heap.add(nums[cnt]);
        //     cnt++;
        // }

        //res[index] = heap.peek();
        return res;
    }
}

 

posted @ 2020-09-17 12:57  3KBLACK  阅读(83)  评论(0)    收藏  举报