zoukankan      html  css  js  c++  java
  • 0239-leetcode算法实现之滑动窗口最大值-sliding-window-maximum-python&golang实现

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

    来源:力扣(LeetCode)
    链接:https://leetcode-cn.com/problems/sliding-window-maximum

    python

    # 0239.滑动窗口最大值
    # https://leetcode-cn.com/problems/sliding-window-maximum/solution/shuang-xiang-dui-lie-jie-jue-hua-dong-chuang-kou-2/
    class Solution:
        def maxSlidingWindow(self, nums: [int], k: int) -> [int]:
            if len(nums) < 2:
                return nums
            from collections import deque
            queue = deque()
            res = [0] * (len(nums)-k+1) # len=len(nums)-k+1
            for i in range(len(nums)):
                # 保证从大到小, 如果前面数小则依次弹出,直到满足条件
                while queue and nums[queue[-1]] < nums[i]:
                    queue.pop()
                queue.append(i)
                # 判断队头是否有效
                if queue[0] <= i-k:
                    queue.popleft()
                # 当窗口长度为k时,保存当前窗口的max值
                if i+1 >= k:
                    res[i+1-k] = nums[queue[0]]
            return res
    

    golang

    package main
    
    // 封装单调队列
    type MyQueue struct {
    	queue []int
    }
    
    func NewMyQueue() *MyQueue {
    	return &MyQueue{
    		queue: make([]int, 0),
    	}
    }
    
    func (m *MyQueue) Front() int {
    	return m.queue[0]
    }
    
    func (m *MyQueue) Back() int {
    	return m.queue[len(m.queue)-1]
    }
    
    func (m *MyQueue) Empty() bool {
    	return len(m.queue) == 0
    }
    
    func (m *MyQueue) Push(val int) {
    	for !m.Empty() && val > m.Back() {
    		m.queue = m.queue[:len(m.queue)-1]
    	}
    	m.queue = append(m.queue, val)
    }
    
    func (m *MyQueue) Pop(val int) {
    	if !m.Empty() && val == m.Front() {
    		m.queue = m.queue[1:]
    	}
    }
    
    // https://leetcode-cn.com/problems/sliding-window-maximum/solution/dai-ma-sui-xiang-lu-dai-ni-gao-ding-zhan-y82r/
    func maxSlidingWindow(nums []int, k int) []int {
    	queue := NewMyQueue()
    	length := len(nums)
    	res := make([]int, 0)
    	// 将前k个元素放入队列
    	for i := 0; i < k; i++ {
    		queue.Push(nums[i])
    	}
    	// 记录前k个元素的max
    	res = append(res, queue.Front())
    
    	for i := k; i < length; i++ {
    		// 滑动窗口移除最前面的元素
    		queue.Pop(nums[i-k])
    		// 滑动窗口添加最后面的元素
    		queue.Push(nums[i])
    		// 记录最大值
    		res = append(res, queue.Front())
    	}
    	return res
    }
    
    
  • 相关阅读:
    iOS UI(布局)约束是什么?view1.attr1 = view2.attr2 * multiplier + constant
    编程范式-声明式编程
    平庸的投资人和优秀的投资人差在哪儿
    编程语言的发展趋势及未来方向
    再读:编程语言的发展趋势及未来方向
    编程语言的发展趋势:声明式动态并发
    命令式编程与声明式编程
    声明式(编程)语言是解释型语言
    DSL-领域特定语言(英语:domain-specific language、DSL)
    声明式编程
  • 原文地址:https://www.cnblogs.com/davis12/p/15531784.html
Copyright © 2011-2022 走看看