LeetCode239.滑動窗口的最大值 | 刷題打卡

題目描述

給你一個整數數組 nums，有一個大小爲 k 的滑動窗口從數組的最左側移動到數組的最右側。你只能夠看到在滑動窗口內的 k 個數字。滑動窗口每次只向右移動一位。

返回滑動窗口中的最大值。

示例 1：

輸入：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
複製代碼

示例 2：

輸入：nums = [1], k = 1
輸出：[1]
複製代碼

示例 3：

輸入：nums = [1,-1], k = 1
輸出：[1,-1]
複製代碼

示例 4：

輸入：nums = [9,11], k = 2
輸出：[11]
複製代碼

示例 5：

輸入：nums = [4,-2], k = 2
輸出：[4]
複製代碼

提示：

• 1 <= nums.length <= 105
• -104 <= nums[i] <= 104
• 1 <= k <= nums.length

思路講解

這道題主要有兩種方法，第一種方法就是利用擴展運算符，將窗口內的值都提取出來，而後用Math.max來尋找最大值，而後把這個最大值添加到result數組中。可是這種方法可能會超出時間限制。

第二種方法就是使用一個雙端隊列（隊列兩面均可進出），用於存儲處於窗口中的值的下標，保證窗口頭部元素永遠是窗口最大值

當前進入的元素下標 - 窗口頭部元素的下標 >= k 頭部元素移出隊列

若是當前數字大於隊列尾，則刪除隊列尾，直到當前數字小於等於隊列尾，或者隊列空 （保證窗口中左側的值均大於當前入隊列的值，這樣作能夠保證當下次循環窗口頭部的元素出隊後，窗口頭部元素仍然爲最大值）

隊列元素入隊

第k次遍歷後開始向結果中添加最大值

AC代碼

方法一：

var maxSlidingWindow = function(nums, k) {
    if (k <= 1) return nums;
    const res = [];
    for (let i = 0; i < nums.length - k + 1; ++i) {
        res.push(Math.max(...nums.slice(i, i + k)));
    }
    return res;

};
複製代碼

方法二：

var maxSlidingWindow = function (nums, k) {
    const window = [];
    const result = [];
    for (let i = 0; i < nums.length; i++) {
      if (i - window[0] > k - 1) {
        window.shift();
      }
      let j = window.length - 1;
      while (j >= 0 && nums[window[j]] <= nums[i]) {
        j--;
        window.pop();
      }
      window.push(i);
      if (i >= k - 1) {
        result.push(nums[window[0]]);
      }
    }
    return result;
  };
複製代碼

總結

第二種解法就是利用了一個雙端隊列，保證在window隊列中的頭元素必定是這個隊列中最大的元素，新加入的元素若是比前面的元素大，就把隊列中的元素都pop出來。窗口每向後移動一位，就把隊列的頭元素push到result數組中。

本文正在參與「掘金 2021 春招闖關活動」, 點擊查看 活動詳情