LeetCode 239 滑動窗口最大值

題目

題目地址

給定一個數組 nums，有一個大小爲 k 的滑動窗口從數組的最左側移動到數組的最右側。你只可以看到在滑動窗口內的 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

使用雙端隊列解決

public static void main(String[] args) {
        int[] origin = {1, 3, -1, -3, 5, 3, 6, 7};
        int k = 3;
        int[] result = maxSlidingWindow(origin, k);
        for (int i : result) {
            System.out.println(i);
        }
    }

    /**
     * 滑動窗口最大值
     *
     * @param nums 原始數組
     * @param k      窗口長度
     * @return
     */
    private static int[] maxSlidingWindow(int[] nums, int k) {
        // 雙端隊列，作爲窗口滑動過程中獲取最大值，存儲數組下標
        Deque<Integer> window = new ArrayDeque<>();
        // 保存結果
        List<Integer> resultTemp = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
            // 開始滑動（0，1，2此時還不需要滑動，因爲窗口大小是3，從下標3開始，纔開始向右滑動），
            // 同時如果窗口中的head元素則彈出最左側的第一個元素，即head
            if (i >= k && window.peekFirst() <= (i - k)) {
                window.pollFirst();
            }
            // window中head只能是當前最大的數字下表，新進入滑動串口的值從後依次向前比，
            // 比他小的都要移除，直到沒有沒有比他小的，則放入窗口的tail中
            while (!window.isEmpty() && nums[window.peekLast()] <= nums[i]) {
                window.pollLast();
            }
            window.add(i);
            // 從下標2包括2開始，將每個窗口內的最大值記錄結果集中
            if (i >= k - 1) {
                resultTemp.add(nums[window.getFirst()]);
            }
        }
        int[] result = new int[resultTemp.size()];
        for (int i = 0; i < resultTemp.size(); i++) {
            result[i] = resultTemp.get(i);
        }
        return result;
    }