给你一个整数数组 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) 链接:leetcode-cn.com/problems/sl… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
package sliding_window
import "container/heap"
type Num struct {
idx int
num int
}
func maxSlidingWindow(nums []int, k int) []int {
queue := []Num{}
maxs := make([]int, len(nums))
for nIdx := 0; nIdx < len(nums); nIdx += 1 {
max := nums[nIdx]
qIdx := len(queue)
for ; qIdx > 0; qIdx -= 1 {
if queue[qIdx-1].num > max {
break
}
}
queue = queue[0:qIdx]
queue = append(queue, Num{idx: nIdx, num: nums[nIdx]})
if nIdx-queue[0].idx >= k {
queue = queue[1:]
}
maxs[nIdx] = queue[0].num
}
maxs = maxs[k-1:]
return maxs
}
// 堆解法
var globalNums []int
type sorter []int
func (s sorter) Less(i, j int) bool {
return globalNums[s[i]] > globalNums[s[j]]
}
func (s sorter) Swap(i, j int) {
s[i], s[j] = s[j], s[i]
}
func (s sorter) Len() int {
return len(s)
}
func (s *sorter) Pop() interface{} {
n := len(*s)
val := (*s)[n-1]
*s = (*s)[:n-1]
return val
}
func (s *sorter) Push(val interface{}) {
*s = append(*s, val.(int))
}
func maxSlidingWindow2(nums []int, k int) []int {
globalNums = nums
s := sorter(make([]int, k))
for i, _ := range s {
s[i] = i
}
heap.Init(&s)
maxs := []int{nums[s[0]]}
for i := k; i < len(nums); i++ {
heap.Push(&s, i)
for s[0] <= i-k {
heap.Pop(&s)
}
maxs = append(maxs, nums[s[0]])
}
return maxs
}