python中两个栈实现一个队列

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()