sisterAn / JavaScript-Algorithms

基础理论+JS框架应用+实践,从0到1构建整个前端算法体系
5.5k stars 633 forks source link

leetcode239:滑动窗口最大值问题 #33

Open sisterAn opened 4 years ago

sisterAn commented 4 years ago

给定一个数组 nums 和滑动窗口的大小 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

提示:

你可以假设 k 总是有效的,在输入数组不为空的情况下,1 ≤ k ≤ 输入数组的大小。

附赠leetcode地址:leetcode

7777sea commented 4 years ago
var maxSlidingWindow = function(nums, k) {

    let max = [];  //存放最大值

    for(let i=0;i<nums.length;i++){
        const _ = nums.slice(i,i + k)
        if(_.length === k){
            max.push(Math.max(..._))
        }

    }

    return max
};
loknum commented 4 years ago
var maxSlidingWindow = function (nums, k) {
    let result = [];
    for (let i = 0; i <= nums.length - k; i++) {
        let item = Math.max(...nums.slice(i, k + i))
        result.push(item);
    }
    return result;
};
2384830985 commented 4 years ago

var maxSlidingWindow = function(nums, k) { let axx = [] let fun = function(){ let arr = nums.slice(0,k) axx.push(Math.max(...arr)) } fun() while(nums.length>k){ nums.shift() fun() } return axx };

zhangxiaos commented 4 years ago
function getMaxSlidingWindow(nums, k) {
  let res = [Math.max(...nums.splice(0, k))]

  if (nums.length === k) {
    return res
  }

  for (let i = 0, len = nums.length; i < len; i++) {
    let curMax = res[res.length - 1];
    console.log(res, curMax)

    curMax = curMax >= nums[i] ? curMax : nums[i]
    res.push(curMax)
  }

  return res
}
zhaojinzhao commented 4 years ago

nums.slice(0, 1 - k).map((v, index) => Math.max(...nums.slice(index, index + k)));

Ttomsjson commented 4 years ago

var ss = function(num,k) { if(!Array.isArray(num) || k <= 0) return

let temp = [], max = 0;

for(var dd = 0; dd < num.length; dd++ ) {
    temp.unshift(num[dd])
    if(dd >= k) temp.length = k
    if(temp.length >= k) {
       max = Math.max.apply(null,temp)
       console.log(max,temp)   
    }
}
return max

} var aa = [1,3,-1,-3,5,3,6,7]; var k = 3; ss(aa,k)

teal-front commented 4 years ago
function slideWindow(arr, k) {
    let i = 0
    let ret = []
    while(i <= arr.length - k) {
        let max = Math.max.apply(Math, arr.slice(i, i + k))
        ret.push(max)
        i++
    }
    return ret
}
sisterAn commented 4 years ago

解答一:暴力解法

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

复杂度分析:

时间复杂度:O(n*k)

空间复杂度:O(n)

解答二:优化:双端队列

解题思路: 使用一个双端队列存储窗口中值的 索引 ,并且保证双端队列中第一个元素永远是最大值,那么只需要遍历一次 nums,就可以取到每次移动时的最大值。

代码实现:

const maxSlidingWindow = function (nums, k) {
    const deque = []
    const result = []
    for (let i = 0; i < nums.length; i++) {
        // 把滑动窗口之外的踢出
        if (i - deque[0] >= k) {
            deque.shift()
        }
        while (nums[deque[deque.length - 1]] <= nums[i]) {
            deque.pop()
        }
        deque.push(i)
        if (i >= k - 1) {
            result.push(nums[deque[0]])
        }
    }
    return result
}

复杂度分析:

leetcode

zjhlaap commented 4 years ago
function maxSlidingWindow(nums, k) {
  const half = Math.ceil(nums.length / 2) + 1;
  const result = [], result2 = [];
  let arr = [], reverseArr = [];
  for (let i = 0; i < half; ++i) {
   arr.push(nums[i])
   reverseArr.push(nums[nums.length - i - 1])
   if(i >= k - 1) {
     result.push(Math.max(...arr))
     result2.push(Math.max(...reverseArr))
     arr.shift()
     reverseArr.shift()
     }
    }
   if (nums.lenght % 2 !== 0) result2.pop(); 
  return result.concat(result2.reverse());
}
AnranS commented 3 years ago
var maxSlidingWindow = function (nums, k) {
    // window存储下标
    let window = [], res = [];

    for (let i = 0; i < nums.length; i++) {
        const num = nums[i];
        // 判断窗口是不是超出长度
        if (window[0] <= i - k) {
            window.shift();
        }
        // 从右开始比较,当当前元素大于队列底元素时,直接弹出
        while(num>nums[window[window.length - 1]]) {
            window.pop();
        }
        window.push(i);
        // 当在窗口内时,加入元素
        if (i >= k - 1) {
            // 加入队顶元素
            res.push(nums[window[0]]);
        }
    }
    return res;
};
azl397985856 commented 3 years ago

题目地址(239. 滑动窗口最大值)

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

题目描述

给定一个数组 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
 

提示:

1 <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4
1 <= k <= nums.length

前置知识

公司

思路

符合直觉的想法是直接遍历 nums, 然后然后用一个变量 slideWindow 去承载 k 个元素, 然后对 slideWindow 求最大值,这是可以的,遍历一次的时间复杂度是 $N$,k 个元素求最大值时间复杂度是 $k$, 因此总的时间复杂度是 O(n * k).代码如下:

JavaScript:

var maxSlidingWindow = function (nums, k) {
  // bad 时间复杂度O(n * k)
  if (nums.length === 0 || k === 0) return [];
  let slideWindow = [];
  const ret = [];
  for (let i = 0; i < nums.length - k + 1; i++) {
    for (let j = 0; j < k; j++) {
      slideWindow.push(nums[i + j]);
    }
    ret.push(Math.max(...slideWindow));
    slideWindow = [];
  }
  return ret;
};

Python3:

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        if k == 0: return []
        res = []
        for r in range(k - 1, len(nums)):
            res.append(max(nums[r - k + 1:r + 1]))
        return res

但是如果真的是这样,这道题也不会是 hard 吧?这道题有一个 follow up,要求你用线性的时间去完成。

其实,我们没必须存储窗口内的所有元素。 如果新进入的元素比前面的大,那么前面的元素就不再有利用价值,可以直接移除。这提示我们使用一个单调递增栈来完成。

但由于窗口每次向右移动的时候,位于窗口最左侧的元素是需要被擦除的,而栈只能在一端进行操作。

而如果你使用数组实现,就是可以在另一端操作了,但是时间复杂度仍然是 $O(k)$,和上面的暴力算法时间复杂度一样。

因此,我们考虑使用链表来实现,维护两个指针分别指向头部和尾部即可,这样做的时间复杂度是 $O(1)$,这就是双端队列。

因此思路就是用一个双端队列来保存接下来的滑动窗口可能成为最大值的数

具体做法:

  1. 一种是已经超出窗口范围了,比如我遍历到第 4 个元素,k = 3,那么 i = 0 的元素就不应该出现在双端队列中了 具体就是索引大于 i - k + 1的元素都应该被清除

  2. 小于当前元素都没有利用价值了,具体就是从后往前遍历(双端队列是一个递减队列)双端队列,如果小于当前元素就出队列

经过上面的分析,不难知道双端队列其实是一个递减的一个队列,因此队首的元素一定是最大的。用图来表示就是:

关键点解析

代码

JavaScript:

JS 的 deque 实现我这里没有写, 大家可以参考 collections/deque

var maxSlidingWindow = function (nums, k) {
  // 双端队列优化时间复杂度, 时间复杂度O(n)
  const deque = []; // 存放在接下来的滑动窗口可能成为最大值的数
  const ret = [];
  for (let i = 0; i < nums.length; i++) {
    // 清空失效元素
    while (deque[0] < i - k + 1) {
      deque.shift();
    }

    while (nums[deque[deque.length - 1]] < nums[i]) {
      deque.pop();
    }

    deque.push(i);

    if (i >= k - 1) {
      ret.push(nums[deque[0]]);
    }
  }
  return ret;
};

复杂度分析

Python3:

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        q = collections.deque() # 本质就是单调队列
        ans = []
        for i in range(len(nums)):
            while q and nums[q[-1]] <= nums[i]: q.pop() # 维持单调性
            while q and i - q[0] >= k: q.popleft() # 移除失效元素
            q.append(i)
            if i >= k - 1: ans.append(nums[q[0]])
        return ans

复杂度分析

扩展

为什么用双端队列

因为删除无效元素的时候,会清除队首的元素(索引太小了)或者队尾(元素太小了)的元素。 因此需要同时对队首和队尾进行操作,使用双端队列是一种合乎情理的做法。

大家对此有何看法,欢迎给我留言,我有时间都会一一查看回答。更多算法套路可以访问我的 LeetCode 题解仓库:https://github.com/azl397985856/leetcode 。 目前已经 37K star 啦。 大家也可以关注我的公众号《力扣加加》带你啃下算法这块硬骨头。

JohnApache commented 3 years ago

这一题用了两种方案,

第一种方案算是暴力写法,也是大多数人常规思路

第二种方案是用了 基于队列的方式 其实,我们没必须存储窗口内的所有元素。 如果新进入的元素比前面的大,那么前面的元素就不再有利用价值,可以直接移除,这样队列对应的值其实就是一个单调递减的队列, 暴力写法 思路清晰,但是代码时间复杂度高 Om*n

const maxSlidingWindow = (nums: number[], k: number) => {
    const result = [];
    for (let i = 0; i < nums.length - k + 1; i++) {
        const tmpArr = [];
        for (let j = i; j < i + k; j++) {
            tmpArr.push(nums[j]);
        }
        result.push(Math.max(...tmpArr));
    }
    return result;
};

队列 队列保存的是在原数组的索引位置,是为了方便控制当窗口移动时,需要删除索引小于 i - k 后的数字,然后每次存储新的索引前队列中推出所有比当前数字小的值的索引,这样每次循环 在 i >= k -1 时存储队列的第一个值索引对应的数字,最终返回结果即可,

我这里用双向链表自己实现了一个方便自己操作的首位数据的队列数据结构

class Node {
    public next: Node | undefined;
    public prev: Node | undefined;
    constructor (public element: number) {}
}

class LinkedList {
    public head: Node | undefined;
    public tail: Node | undefined;
    public length: number = 0;
    append (element: number) {
        const node = new Node(element);
        if (!this.head || !this.tail) {
            this.head = node;
            this.tail = node;
            this.length++;
            return;
        }
        this.tail.next = node;
        node.prev = this.tail;
        this.tail = node;
        this.length++;
    }
    shift () {
        if (!this.head || !this.tail) return;
        const next = this.head.next;
        if (!next) {
            this.head = undefined;
            this.tail = undefined;
            this.length--;
            return;
        }
        next.prev = undefined;
        this.head = next;
        this.length--;
    }
    pop () {
        if (!this.head || !this.tail) return;
        const prev = this.tail.prev;
        if (!prev) {
            this.head = undefined;
            this.tail = undefined;
            this.length--;
            return;
        }
        prev.next = undefined;
        this.tail = prev;
        this.length--;
    }
    top () {
        return this.head;
    }
    bottom () {
        return this.tail;
    }
}

const maxSlidingWindow = (nums: number[], k: number) => {
    const queue = new LinkedList(); // 队列
    const result = [];
    for (let i = 0; i < nums.length; i++) {
        const num = nums[i];
        let topNode = queue.top();
        if (topNode && (i - topNode.element >= k)) {
            queue.shift();
        }

        let bottomNode = queue.bottom();
        while (bottomNode && nums[bottomNode.element] < num) {
            queue.pop();
            bottomNode = queue.bottom();
        }
        queue.append(i);
        topNode = queue.top();
        if (topNode && i >= k - 1) {
            result.push(nums[topNode.element]);
        }
    }
    return result;
};
console.log(maxSlidingWindow([1], 1))
console.log(maxSlidingWindow([1,-1], 1))
console.log(maxSlidingWindow([9,11], 2))
console.log(maxSlidingWindow([4,-2], 2))
console.log(maxSlidingWindow([1,3,-1,-3,5,3,6,7], 3))
console.log(maxSlidingWindow([5,1,2,1,5], 3))
// 输出
// [ 1 ]
// [ 1, -1 ]
// [ 11 ]
// [ 4 ]
// [ 3, 3, 5, 5, 6, 7 ]
// [ 5, 2, 5 ]