本文已参与「新人创作礼」活动, 一起开启掘金创作之路。
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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();
*/