刷题笔记day13-栈和队列part03

239. 滑动窗口最大值

func maxSlidingWindow(nums []int, k int) []int {
    // 思路：用一个单调队列（入口方向看，是递增的）
    // 往队列中push的时候，只要比入口大，就弹出入口元素，直到小于前面的值时，插入当前值。
    // 滑动窗口的时候，删除尾值，如果比队列中的出口元素一样，则删除，否则就不用操作
    // 那么que的队头一直是最大值。
    
    que := &Queue{}
    result := []int{}
    // 先push进去k个
    
    for i := 0; i < k && i < len(nums); i++ {
        que.push(nums[i])
    }
    result = append(result, que.front())
    for i := k; i < len(nums); i++ {
        // 移除队尾v，
        v := nums[i - k]
        // 然后比较v和队列的队头的值，等于则移除
        // 因为队头是最大的
        que.pop(v) // 删除旧元素
        que.push(nums[i]) //追加新元素
        result = append(result, que.front())
    }
    return result
}

type Queue struct {
    data []int
    length int
}

func (q *Queue) push(v int) {
    // push，入口的比 v 小就删除
    for !q.empty() && v > q.back() {
        q.pop_back()
    }
    q.push_back(v)
}

func (q *Queue) push_back(v int) {
    q.data = append(q.data, v)
    q.length++
}

func (q *Queue) back() (v int) {
    return q.data[q.length - 1]
}

func (q *Queue) pop(v int) {
    if v == q.front() {
        q.pop_front()
    }
}

func (q *Queue) front() (v int) {
    return q.data[0]
}

func (q *Queue) pop_front() {
    q.data = q.data[1:]
    q.length--
}

func (q *Queue) pop_back() {
    q.data = q.data[:q.length - 1]
    q.length--
}

func (q *Queue) empty() bool {
    return q.length <= 0
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

8.前 K 个⾼频元素