python中两个栈实现一个队列
- 栈:先入后出
- 队列:先入先出
- 思想:队列A是入栈的,队列B是出栈的,入栈时,直接进入A,出栈时,先判断是队列B否有元素,由于队列B中如果没有元素会发生索引错误,所以应该先将A中所有的元素压倒B里面,再pop最上面一个元素,如果B中有就直接pop出,就可以了
class Method:
# 初始化两个栈
def __init__(self):
self.stackA = []
self.stackB = []
def push(self, node):
# write code here
self.stackA.append(node)
def pop(self):
# 若压入的栈中有元素则直接弹出
if self.stackB:
self.stackB.pop()
# 若A中没有元素则返回None
elif not self.stackA:
return None
# 若A中有元素,则统一压入B中进行弹出
else:
while self.stackA:
self.stackB.append(self.stackA.pop())
return self.stackB.pop()