滑动窗口最大值的golang实现
程序员文章站
2023-03-31 18:06:22
给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口 k 内的数字。滑动窗口每次只向右移动一位。 返回滑动窗口最大值 其实这道题就是求给定数组中获取全部K个连续元素中最大值的集合 首先我们可能会遇到三中情况 当原始数组为空的,那就直接返回一个空 ......
给定一个数组 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
其实这道题就是求给定数组中获取全部k个连续元素中最大值的集合
首先我们可能会遇到三中情况
- 当原始数组为空的,那就直接返回一个空数组
- 如果原始数组的长度与给定的k是一样的,那么就直接去原始数组的最大值即可
- 如果原始数组的长度大于k,那么我们就要求每个连续子数组的最大值了
对于第一种情况就非常简单了
var result []int //如果切片长度为0的话,那就直接返回空切片 if len(nums) == 0 { return result }
对于第二种情况:
//如果切片长度与k一样 if len(nums) == k { result = append(result, getmax(nums)) return result }
对于第三种情况:
var windowslice []int index := 0 for i := k; i <= len(nums); i++ { windowslice = nums[index:i] result = append(result, getmax(windowslice)) index++ } return result
这里每次都获取连续k个元素进行比较
全部代码:
package main import "fmt" func maxslidingwindow(nums []int, k int) []int { var result []int //如果切片长度为0的话,那就直接返回空切片 if len(nums) == 0 { return result } //如果切片长度与k一样 if len(nums) == k { result = append(result, getmax(nums)) return result } var windowslice []int index := 0 for i := k; i <= len(nums); i++ { windowslice = nums[index:i] result = append(result, getmax(windowslice)) index++ } return result } func getmax(windowslice []int) int { max := windowslice[0] for i := 0; i < len(windowslice); i++ { if windowslice[i] >= max { max = windowslice[i] } } return max } func main() { nums := []int{1, 3, -1, -3, 5, 3, 6, 7} fmt.println(maxslidingwindow(nums, 3)) // nums := []int{} // fmt.println(maxslidingwindow(nums, 0)) // nums := []int{1, -1} // fmt.println(maxslidingwindow(nums, 1)) }
附上老师的解法,但是我看不懂!
func maxslidingwindow(nums []int, k int) []int { result := []int{} if len(nums) == 0 { return result } window := []int{} for i, x := range nums { if i >= k && window[0] <= i-k { window = window[1:] } for len(window) != 0 && nums[window[len(window)-1]] <= x { window = []int{} } window = append(window, i) if i >= k-1 { result = append(result, nums[window[0]]) } } return result }
上一篇: Python练手例子(7)