本題來自左神《程序員面試代碼指南》“生成窗口最大值數組”題目。

題目

有一個整型數組 arr 和一個大小為 w 的窗口從數組的最左邊滑到最右邊，窗口每次向右邊滑一個位置。

例如，數組為[4,3,5,4,3,3,6,7]，窗口大小為3時：

窗口數組 最大值 [4 3 5] 4 3 3 6 7 54 [3 5 4] 3 3 6 7 54 3 [5 4 3] 3 6 7 54 3 5 [4 3 3] 6 7 44 3 5 4 [3 3 6] 7 64 3 5 4 3 [3 6 7] 7

如果數組長度為 n，窗口大小為 w，則一共產生 n-w+1 個窗口的最大值。

請實現一個函數，

• 輸入：整型數組 arr，窗口大小為 w；
• 輸出：一個長度為 n-w+1 的數組 res，res[i] 表示每一種窗口狀態下的最大值。

以本題為例，結果應該返回[5,5,5,4,6,7]

題解

思路總結：每一次從 隊尾 將 小于等于 當前元素的彈出

代碼

import java.util.LinkedList;public class Main {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]) { // 一直彈出隊尾位置，直到隊尾對應元素大于當前元素為止qmax.pollLast();}qmax.addLast(i); // 放入當前元素if (qmax.peekFirst() == i - w) { // 如果qmax對頭下標過期（已在當前窗口覆蓋范圍之外），則彈出對頭下標qmax.pollFirst();}if (i >= w - 1) { // 如果當前i及其以前的元素滿足一個窗口大小，則記錄當前窗口的最大值res[index++] = arr[qmax.peekFirst()];}}return res;}// for testpublic 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));} }

運行結果

5 5 5 4 6 7 超強干貨來襲 云風專訪：近40年碼齡，通宵達旦的技術人生

總結

以上是生活随笔為你收集整理的左神算法：生成窗口最大值数组（Java版）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。