生成窗口最大的数值组(栈和队列) 笔记

2018-09-30  本文已影响0人  王古

有一个整型数组arr和一个大小为w 的窗口从数组的最左滑到最右边,窗口每次向右边滑一个位置,求每次滑动后的窗口的最大值。

时间复杂度为O(N)

生成双端队列qmax,qmax中存放数组arr的下标

import java.util.LinkedList;

public class Problem_07_SlidingWindowMaxArray {

    public static int[] getMaxWindow(int[] arr, int w) {
        if (arr == null || w < 1 || arr.length < w) {
            return null;
        }
        LinkedList<Integer> qmax = new LinkedList<Integer>();
        int[] res = new int[arr.length - w + 1];
        int index = 0;
        for (int i = 0; i < arr.length; i++) {                       
            while (!qmax.isEmpty() && arr[qmax.peekLast()] <= arr[i]) {  //peeklast() 检索但是不移除最后一个元素
                qmax.pollLast(); //检索并移除最后一个元素
            }
            qmax.addLast(i);
            if (qmax.peekFirst() == i - w) {
                qmax.pollFirst();
            }
            if (i >= w - 1) {
                res[index++] = arr[qmax.peekFirst()];
            }
        }
        return res;
    }

    // for test
    public static void printArray(int[] arr) {
        for (int i = 0; i != arr.length; i++) {
            System.out.print(arr[i] + " ");
        }
        System.out.println();
    }

    public static void main(String[] args) {
        int[] arr = { 4, 3, 5, 4, 3, 3, 6, 7 };
        int w = 3;
        printArray(getMaxWindow(arr, w));

    }

} 

结果:


image.png
上一篇下一篇

猜你喜欢

热点阅读