滑动窗口的最大值

137 阅读1分钟

滑动窗口的最大值

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

例如,如果输入数组[2, 3, 4, 2, 6, 2, 5, 1]及滑动窗口的大小3,那么一共存在6个滑动窗口,它们的最大值分别为[4, 4, 6, 6, 6, 5]。

注意:

数据保证k大于0,且k小于等于数组长度。

样例
输入:[2, 3, 4, 2, 6, 2, 5, 1] , k=3
输出: [4, 4, 6, 6, 6, 5]

优先级队列

时间复杂度O(n)

class Solution {
    public int[] maxInWindows(int[] nums, int k) {
        if(nums.length == 0 || nums == null) {
            return nums;        
        }
        LinkedList<Integer> list = new LinkedList<Integer>();
        Queue<Integer> q = new PriorityQueue<Integer>(k,(a,b) -> b - a);
        for(int i = 0;i < k;i++) {
            q.offer(nums[i]);
        }
        for(int i = 0;i < nums.length - k + 1;i++) {
            list.add(q.peek());
            q.remove(nums[i]);
            if(i != nums.length - k) {
                q.offer(nums[i + k]);
            }
        }
        return list.stream().mapToInt(a -> a.intValue()).toArray();
    }
}