欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

滑动窗口最大值的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
}