刷题笔记day13-栈和队列part03

最新推荐文章于 2024-08-24 16:43:50 发布

爱飞的蚂蚁

最新推荐文章于 2024-08-24 16:43:50 发布

阅读量82

点赞数

文章标签：笔记算法

本文链接：https://blog.csdn.net/weixin_39646458/article/details/134255006

版权

ACM刷题专栏收录该内容

19 篇文章 0 订阅

订阅专栏

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
}