0239-leetcode算法实现之滑动窗口最大值-sliding-window-maximum-python&golang实现

发布时间:2022-06-30 发布网站:脚本宝典
脚本宝典收集整理的这篇文章主要介绍了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
}

脚本宝典总结

以上是脚本宝典为你收集整理的0239-leetcode算法实现之滑动窗口最大值-sliding-window-maximum-python&golang实现全部内容,希望文章能够帮你解决0239-leetcode算法实现之滑动窗口最大值-sliding-window-maximum-python&golang实现所遇到的问题。

如果觉得脚本宝典网站内容还不错,欢迎将脚本宝典推荐好友。

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。
标签: