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