两个栈实现一个队列

思想：两个栈stack1,stack2;元素严格从一个栈进入，另一个栈出。

class Solution:
    def __init__(self):
        self.stack1 = []
        self.stack2 = []

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

    def pop(self):
        #stack2不为空，就将stack2中的元素弹出；若为空，就将stack1中所有的元素输入到stack2中，                
        #然后再将stack2中的元素弹出
        if self.stack2 == []:
            for i in range(len(self.stack1)):
                self.stack2.append(self.stack1.pop())
        return self.stack2.pop()


s = Solution()
s.push(5)
s.push(3)
s.push(4)
ele = s.pop()
print(len(s.stack2))
print(ele)