LeetCode滑动窗口最大值&队列最大值

队列的最大值

题目

请定义一个队列并实现函数 max_value 得到队列里的最大值，要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。

若队列为空，pop_front 和 max_value 需要返回 -1

Code

// 队列最大值
type MaxQueue struct {
    doubleQue []int
    maxQue []int
}


func Constructor() MaxQueue {
    return MaxQueue {
        make([]int, 0),
        make([]int, 0),
    }
}


func (this *MaxQueue) Max_value() int {
    if len(this.maxQue) == 0 {
        return -1
    }
    return this.maxQue[0]
}


func (this *MaxQueue) Push_back(value int)  {
    this.doubleQue = append(this.doubleQue, value)
    for len(this.maxQue)!= 0 && this.maxQue[len(this.maxQue)-1] < value {
        this.maxQue = this.maxQue[:len(this.maxQue)-1]
    }
     this.maxQue = append(this.maxQue, value)
}


func (this *MaxQueue) Pop_front() int {
    n := -1
    if len(this.doubleQue) != 0 {
        n = this.doubleQue[0]
        this.doubleQue = this.doubleQue[1:]
        if len(this.maxQue)!= 0 && n == this.maxQue[0] {
            this.maxQue = this.maxQue[1:]
        }
    }
    return n
}


/**
 * Your MaxQueue object will be instantiated and called as such:
 * obj := Constructor();
 * param_1 := obj.Max_value();
 * obj.Push_back(value);
 * param_3 := obj.Pop_front();
 */

滑动窗口最大值

题目

给定一个数组 nums 和滑动窗口的大小 k，请找出所有滑动窗口里的最大值。

Code

//自定义一个队列，方便操作
type queue []int

func (q *queue) popFirst(){
	*q = (*q)[1:]
}
func (q *queue) popLast() {
	*q = (*q)[:len(*q)-1]
}
func (q *queue) add(a int){
	*q = append(*q, a)
}
func (q *queue) getFirst()int{
	res := (*q)[0]
	return res
}

//双端队列法
//这道题不能用queue的长度来判断是否元素过期，因为元素动态添加，queue只能存索引
func maxSlidingWindow(nums []int, k int) []int {
	//总体思路：保证queue中的第一个数就是当前最大的，
	var res []int
	if len(nums)<k || k<1{return res}
	var q queue
	// 遍历nums， 
	for i:=0;i<len(nums);i++{
		// 遍历queue，弹出所有比nums[i]小的，然后再添加nums[i]，确保queue的第一个还是最大的
		for j:=len(q)-1;j>=0;j--{
			if nums[q[j]]<=nums[i]{
				q.popLast()
			}else{
				break
			}
		}
		q.add(i)
		
		// 如果queue中的元素有过期的，那么扔掉过期
		if q.getFirst() <= i-k{
			q.popFirst()
		}
		// 如果窗口大小达到k，那么添加
		if i>=k-1{
			res = append(res, nums[q.getFirst()])
		}
	}
	return res
}