面试题59_1:滑动窗口的最大值

2019-11-22  本文已影响0人  繁星追逐

给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个: {[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1}, {2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]}。

/**
     * 使用优先队列控制入队数量,输出最大值
     * @param num
     * @param size
     * @return
     */
    public static ArrayList<Integer> maxInWindows(int [] num, int size)
    {
        ArrayList<Integer> list = new ArrayList<>();
        if (num == null || size > num.length || num.length == 0) return list;
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>(Comparator.reverseOrder());
        int j = 0;
        for (int i=0;i<num.length;i++){
            maxHeap.offer(num[i]);
            if (maxHeap.size() >= size){
                list.add(maxHeap.peek());
                maxHeap.remove(num[j++]);
            }
        }
        return list;
    }
    /**
     * 方法2: 使用双端队列,存放下标
     * 用一个双端队列,队列第一个位置保存当前窗口的最大值,当窗口滑动一次
     * 1.判断当前最大值是否过期
     * 2.新增加的值从队尾开始比较,把所有比他小的值丢掉
     */
    public static ArrayList<Integer> maxInWindow2(int[] num, int size) {
        ArrayList<Integer> list = new ArrayList<>();
        if (num == null || num.length < size || size <= 0) return list;
        Deque<Integer> deque = new LinkedList<>();
        for (int i = 0; i < num.length; i++) {
            //循环保证比当前值小的都出来
            while (!deque.isEmpty() && num[i] >= num[deque.peekLast()]) deque.pollLast();
            //记录位置不能超过3个大小
            if (!deque.isEmpty() && i - deque.peekFirst() >= size) deque.pollFirst();
            //入队
            deque.offerLast(i);
            //
            //索引值足够大,取出对头元素对应下标的值
            if (i +1 >= size) list.add(num[deque.peekFirst()]);
        }
        return list;
    }

    /**
     * 最简单的轮询方式
     * @param num
     * @param size
     * @return
     */
    public static ArrayList<Integer> maxInWindows1(int [] num, int size)
    {
        ArrayList<Integer> list = new ArrayList<>();
        if (num == null || num.length < size || size <= 0) return list;

        int len = num.length;
        for (int i=0;i<len-size+1;i++){
            int max = num[i];
            int end = i+size;
            for (int j=i;j<end;j++){
                if (max < num[j]){
                    max = num[j];
                }
            }
            list.add(max);
        }
        return list;
    }
上一篇下一篇

猜你喜欢

热点阅读