天天看點

滑動視窗最大值

滑動視窗最大值

給你一個整數數組

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
           
輸入:nums = [1], k = 1
輸出:[1]
           
輸入:nums = [1,-1], k = 1
輸出:[1,-1]
           
輸入:nums = [9,11], k = 2
輸出:[11]
           

題解

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number[]}
 */
var maxSlidingWindow = function(nums, k) {
    const indexGroup = [], maxGroup = [];
    nums.forEach((v,i) => {
        if(i >= k && indexGroup[0] <= i-k) indexGroup.shift();
        while(indexGroup && nums[indexGroup[indexGroup.length-1]] <= v) indexGroup.pop();
        indexGroup.push(i);
        if(i >= k-1) maxGroup.push(nums[indexGroup[0]]);
    })
    return maxGroup;
};

// 普通做法逾時 需要優化的實作方式
// var maxSlidingWindow = function(nums, k) {
//     const maxGroup = [];
//     for(let i=0,n=nums.length; i<=n-k; ++i){
//         maxGroup.push(Math.max(...nums.slice(i, i+k)));
//     }
//     return maxGroup;
// };
           

是猛男就要爆破的方式不管用了,逾時警告,那麼隻能優化做法,實際上對于兩個相鄰的滑動視窗,它們共用着

k-1

個元素,而隻有

1

個元素是變化的,我們可以根據這個特點進行優化。我們可以通過維護一個單調遞減的視窗來實作,當向右移動時左側超出視窗的值彈出,因為需要的是視窗内的最大值,是以隻要保證視窗内的值是遞減的即可,即小于新加入的值全部彈出,最左端即為視窗最大值。首先我們定義一個用來存儲遞減值的下标的視窗,以及存儲最大值的組,之後循環給定的數組,如果目前周遊的數組值下标大于視窗大小并且遞減下标視窗的第一個值是小于目前視窗,即第一個值在目前需要組合的視窗之外,就将其彈出,之後從後向前周遊,如果遞減視窗存在值且其中的值小于即将要加入的值就将其彈出,此時将目前周遊的值的下标加入遞減視窗,最後如果視窗能夠組合成

k

個就開始取最大值即遞減視窗的第一個值,将其加入最大值組,循環結束後傳回即可。

每日一題

https://github.com/WindrunnerMax/EveryDay
           

參考

https://leetcode-cn.com/problems/sliding-window-maximum/