假设一个固定大小为W的窗口,依次划过arr,返回每一次滑出状况的最大值。例如,arr = [4,3,5,4,3,3,6,7], W = 3。返回:[5,5,5,4,6,7]。
采用双端队列,存序号。遍历数组。 1.当双端队列里没值或者双端队列最右边的值小于当前值,则把当前值的序号从右边push到队列里。 2.否则pop最右边的序号,直到符合条件为止。 3.双端队列左边的序号太小,当前序号-左序号>=窗口大小W,需要pop左边的序号。 4.双端队列最右边的值就是最大值。 有代码。
代码用golang编写,代码如下:
package main
import (
"container/list"
"fmt"
)
func main() {
arr := []int{4, 3, 5, 4, 3, 3, 6, 7}
w := 3
ret := getMaxWindow(arr, w)
fmt.Println(ret)
}
func getMaxWindow(arr []int, w int) []int {
arrLen := len(arr)
if arrLen < w || w < 1 {
return nil
}
qmax := list.New().Init()
res := make([]int, arrLen-w+1)
index := 0
for R := 0; R < arrLen; R++ {
for qmax.Len() > 0 && arr[qmax.Back().Value.(int)] <= arr[R] {
qmax.Remove(qmax.Back())
}
qmax.PushBack(R)
if qmax.Front().Value.(int) == R-w {
qmax.Remove(qmax.Front())
}
if R >= w-1 {
res[index] = arr[qmax.Front().Value.(int)]
index++
}
}
return res
}
执行结果如下: