两个栈实现一个队列

#进栈：元素入队列A
#出栈：判断如果队列A只有一个元素，则直接出队。否则，把队A中的元素出队并入队B，直到队A中只有一个元素，再直接出队。为了下一次继续操作，互换队A和队B。
class Solution:
    def __init__(self):
        self.stackA = []  # stackA用来保存进队的元素
        self.stackB = []  # stackB用来保存将要出队的元素

    def push(self, node):
        self.stackA.append(node)

    def pop(self):
        if not self.stackA and not self.stackB:#出栈列表中无元素（元素全部入栈），且入栈列表为空
            return None
        if not self.stackB:  # 当stackB为空的时候将stackA的元素全部逆序压入
            while self.stackA:
                self.stackB.append(self.stackA.pop())
        
        return self.stackB.pop()#尾部开始输出元素

a=[1,2,3,4,5,6]
s=Solution()
while a:
    s.push(a.pop())
# s.gettest()
s.push(88)

print("In_Stack = ",s.stackA)
# print(s.stackB)
print(s.pop())
while s.stackB:
    print(s.pop())