栈实现队列

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

class Queue(object):
    def __init__(self):
        self.stack_a = []
        self.stack_b = []

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

    def pop(self):
        if not self.stack_a:
            while self.stack_a:
                self.stack_b.append(self.stack_a.pop())
        return self.stack_b.pop()

　　

a = [1,2,3,4]
print(a.pop())
print(a)

>>4
>>[1, 2, 3]