队列的最大值
题目
请定义一个队列并实现函数 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
}