【编程题】滑动窗口的最大值

问题描述

给定一个数组和滑动窗口的大小，找出所有滑动窗口里数值的最大值。例如，如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3，那么一共存在6个滑动窗口，他们的最大值分别为{4,4,6,6,6,5}； 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个： {[2,3,4],2,6,2,5,1}， {2,[3,4,2],6,2,5,1}， {2,3,[4,2,6],2,5,1}， {2,3,4,[2,6,2],5,1}， {2,3,4,2,[6,2,5],1}， {2,3,4,2,6,[2,5,1]}。

思路：两端开口的队列

• 我们借助一个两端开口的队列，用来保存有可能是滑动窗口最大值的数字的下标。
• 在存入一个数字的下标之前，首先要判断队列里已有数字是否小于待存入的数字。如果已有数字小于待存入的数字，那么这些数字已经不可能是滑动窗口的最大值，因此它们将会被依次从队列的尾部删除。
• 同时，如果队列头部的数字已经从窗口滑出，那么滑出的数字也需要从队列的头部删除。队列头部的数字从窗口滑出的条件：即将保存的数字的下标 - 队列头部保存的下标 >= 滑动窗口的大小。
• 这样，队列头部的数字就是每次滑动窗口的最大值。

代码实现

function maxInWindows(arr,size){

    if(!arr.length || !size || arr.length < size){
        return [];
    }

    let max = []; // 用来保存所有滑动窗口里数值的最大值
    let queue = []; // 用来保存有可能是滑动窗口最大值的数字的下标

    for(let i=0; i<size; i++){
        // 在存入一个数字的下标之前，首先要判断队列里已有数字是否小于待存入的数字。
        // 如果已有数字小于待存入的数字，那么这些数字已经不可能是滑动窗口的最大值，因此它们将会被依次从队列的尾部删除
        while(queue.length && arr[i] > arr[queue[queue.length-1]]){
            queue.pop();
        }
        queue.push(i); // 每次都存入一个数字的下标
    }

    for(let i=size; i<arr.length; i++){
        max.push(arr[queue[0]]); // 队列头部的数字就是每次滑动窗口的最大值
        while(queue.length && arr[i] > arr[queue[queue.length-1]]){
            queue.pop();
        } 
        // 如果队列头部的数字已经从窗口滑出，那么滑出的数字也需要从队列的头部删除。
        // 队列头部的数字从窗口滑出的条件：即将保存的数字的下标 - 队列头部保存的下标 >= 滑动窗口的大小。
        if(queue.length && queue[0] <= i-size){
            queue.shift();
        }       
        queue.push(i);
    }
    max.push(arr[queue[0]]); // 加入最后一个滑动窗口的最大值

    return max;
}

// 测试
let arr = [2,3,4,2,6,2,5,1];
let size = 3;
console.log(maxInWindows(arr,size)); // [4,4,6,6,6,5]

END