《剑指Offer》
题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
栈:先进后出;队列:先进先出
若要想用两个栈实现队列的过程,可根据栈的特性,即先进后出,让一个栈用来进行存数,计为1号栈,可实现“入队”操作。
当想实现出队操作时,需要实现的效果是把1号栈中最开始存的数,即栈底元素先出,但我们发现当该栈进行取数据时,栈底元素只能最后出来,这时候就可以用另一个栈,计为2号栈,来将1号栈取出的数依次按顺序再存进来,1号栈中的栈底元素就变成了2号栈中的栈顶元素,当从2号栈中进行取数时,就实现了“出队操作”。
class Solution:
def __init__(self):
self.acceptStack = []
self.outputStack = []
def push(self, node):
# write code here
self.acceptStack.append(node)
def pop(self):
# return xx
if self.outputStack == []:
while self.acceptStack:
self.outputStack.append(self.acceptStack.pop())
if self.outputStack != []:
return self.outputStack.pop()
else:
return None
# 可以测试一下
solution = Solution()
solution.push(123)
solution.pop()
总结:由于栈的特殊操作性质,当对元素进行两次的存取时,就相当于实现了一个“逆置换“。