栈与队列之间的相互实现,是面试中的经典试题。
两个栈实现一个队列
入队:元素进栈A
出队:先判断栈B是否为空,为空则将栈A中的元素 pop 出来并 push 进栈B,再栈B出栈,如不为空则栈B直接出栈
复杂度分析:
这样用两个栈实现一个队列,入队的复杂度为O(1),出队的复杂度则变为O(n)。
而直接用 python 的单个列表实现队列,以列表首作为队列尾,则入队用insert,复杂度为O(n),出队用pop,复杂度为O(1)。(列表首,即列表中下标为0的元素)
实现:就以列表作为栈的底层实现,只要保证后进先出的约束就是栈。这里只实现入队和出队两个操作。
class Queue:
def __init__(self):
self.stockA=[]
self.stockB=[]
def push(self, node):
self.stockA.append(node)
def pop(self):
if self.stockB==[]: