239. 滑动窗口最大值

177 阅读1分钟

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