1 题目描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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 次调用
2 解题思路
上述队列声明中可以看出,一个队列包含了两个栈stack1和stack2,因此这道题的意图就是要求我们操作这两个“后进先出”的栈,实现一个“先进先出”的队列CQueue。
我们可以通过一个具体的例子来分析往该队列插入和删除元素的过程。首先插入一个元素a,不妨先把它插入stack1,此时栈中有{a},stack2为空。再压入两个元素b和c,还是插入stack1中,此时stack1中的元素有{a,b,c},其中c位于栈顶,而stack2仍然是空的,如下图所示。
这时候试着从队列中删除一个元素。按照队列先入先出的规则,由于a比b、c先插入队列中,最先别删除的元素应该是a。元素a存储在stack1中,但是并不在栈顶上,因此不能直接进行删除。注意到stack2一直没有被使用过,现在是让stack2发挥作用的时候了。如果我们把stack1中的元素逐个弹出并压入stack2,则元素在stack中的顺序正好和原来在在stack中的顺序相反,因此经过3次弹出stack1和压入stack2的操作之后,stack1为空,而stack2中的元素是{c,b,a},这时候就可以弹出stack2的栈顶a了。此时的stack1为空,而stack2的元素为{b,c},其中b在栈顶。
如果我们还想继续删除队列的头部,剩下的两个元素正好是b和c,b比c早入队列,因此b应该先删除。此时b正好又在栈顶上,因此直接弹出stack2的栈顶即可。在这次弹出操作之后,stack1仍然为空,而stack2中的元素为c。
从上面的分析中我们可以总结出删除一个元素的步骤:如果stack1和stack2为空时,队列为空,返回-1;当stack2不为空时,在stack2的栈顶元素就是最先进入队列的元素,可以弹出。当stack2为空时,我们把stack1中的元素逐个弹出并压入stack2.由于先进入队列的元素被压到stack1的底端,经过弹出和压入操作之后就处于stack2的顶端,又可以直接弹出。
接下来在插入一个元素d。我们还是把它压入stack1,这样会不会有问题呢?我们考虑下一次删除队列的头部stack2不为空,直接弹出他的栈顶元素c。而c的确比d先进入队列,应该在d之前从队列中删除,因此不会出现任何矛盾。
3 代码
class CQueue:
def __init__(self):
self.A=[]
self.B=[]
def appendTail(self, value: int) -> None:
self.A.append(value)
def deleteHead(self) -> int:
if not self.B and not self.A:
return -1
if self.B:
return self.B.pop()
while self.A:
self.B.append(self.A.pop())
while self.B:
return self.B.pop()