剑指 Offer 09. 用两个栈实现队列

85 阅读2分钟

本文已参与「新人创作礼」活动, 一起开启掘金创作之路。

 剑指 Offer 09. 用两个栈实现队列

用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )

示例 1:

输入:
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出:[null,null,3,-1]
示例 2:

输入:
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]
提示:

1 <= values <= 10000
最多会对 appendTail、deleteHead 进行 10000 次调用

思路:
参考:大佬题解
1. 加入队尾 appendTail() 函数: 将数字 val 加入栈 A 即可;
2. 删除队首 deleteHead() 函数,有以下三种情况:

  • 当栈 B 不为空:B中仍有已完成倒序的元素,因此直接返回 B 的栈顶元素。
  • 当栈B为空,栈A不为空:将栈 A 元素全部转移至栈 B 中,实现元素倒序,并返回栈 B 的栈顶元素。
  • 当栈B为空,栈 A也为空: 即两个栈都为空,无元素,因此返回 -1

注意:移动后,记得删除移动前位置的元素(如栈A移动到栈B,记得清除栈A中原有元素)

时间复杂度:
appendTail()函数为O(1) ;deleteHead() 函数在 N 次队首元素删除操作中总共需完成 N 个元素的倒序。

空间复杂度 O(N) :
最差情况下,栈 A 和 B 共保存 N 个元素。

type CQueue struct {
    stack1 []int
	stack2 []int
}


func Constructor() CQueue {
    return CQueue{
		stack1: make([]int, 0),
		stack2: make([]int, 0),
	}
}


func (this *CQueue) AppendTail(value int)  {
    this.stack1 = append(this.stack1, value)
}


func (this *CQueue) DeleteHead() int {
 if len(this.stack2) > 0 {
		res := this.stack2[len(this.stack2)-1] // 先记录栈顶值,再出栈
		this.stack2 = this.stack2[:len(this.stack2)-1] // stack2 pop后,也要删除stack2中被pop的元素
		return res
	} else {
		if len(this.stack1) > 0 { // 2.1: stack2为空,stack1不为空,此时移动stack1中所有元素到stack2中,然后stack2执行pop操作
			for i := len(this.stack1) - 1; i >= 0; i-- { // stack1中的元素全部移到stack2中
				this.stack2 = append(this.stack2, this.stack1[i])
				this.stack1 = this.stack1[:len(this.stack1)-1] // stack1中的元素搬到stack2后,stack1中的元素要删除!!!
			}

			res := this.stack2[len(this.stack2)-1]
			this.stack2 = this.stack2[:len(this.stack2)-1] // stack2 pop后,也要删除stack2中被pop的元素
			return res
		} else { // 2.2 stack2为空,stack1也为空,此时返回-1
			return -1
		}
	}
}


/**
 * Your CQueue object will be instantiated and called as such:
 * obj := Constructor();
 * obj.AppendTail(value);
 * param_2 := obj.DeleteHead();
 */