题目:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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,2,3,4},先入队放到一个栈中--[1,2,3,4],然后将这个栈的元素出栈--[4,3,2,1],再将出栈元素入栈到另外一个栈--[4,3,2,1],最后将这个栈出栈即可[1,2,3,4],这样就实现了出队操作。从[1,2,3,4]入队到[1,2,3,4]出队(先进先出)。
class CQueue:
def __init__(self):
self.stack1 = []
self.stack2 = []
def appendTail(self, value: int) -> None:
self.stack1.append(value) #直接将元素插入
def deleteHead(self) -> int:
if not self.stack2: #当第二个栈为空的时候进行(不为空证明队列中还有元素,出队即可)
if not self.stack1: #第一个栈为空返回-1(没有出栈的元素)
return -1
while self.stack1: #当第一个栈不为空的时候
self.stack2.append(self.stack1.pop())#将第一个栈的栈顶元素加入第二个栈,这样新的栈的栈顶元素就是刚入队的元素。
res = self.stack2.pop() #然后将新的栈顶元素出栈即是队列的出队。
return res
注:python列表:
pop方法是删除并返回指定位置元素,如果未指定位置则默认操作列表最后一个元素。
append方法是在列表尾部添加新的元素,这两个方法搭配正好可以实现入栈(append)和出栈(pop)。