剑指offer-用两个栈实现队列

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

1. push操作：进栈和进队列是一样的，进入栈1；
2. pop操作：如果栈2不为空，就从栈2pop出来；
3. 若栈2为空，就把栈1里的元素全部pop进栈2，之后再从栈2里pop

class Solution(object):
    def __init__(self):
        self.stack1 = []
        self.stack2 = []
    def push(self,node):
        if node:
            self.stack1.append(node)
    def pop(self):
        if self.stack2:
            return self.stack2.pop()
        else:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
            return self.stack2.pop()

测试用例

if __name__ == '__main__':
    s = Solution()
    s.push(1)
    s.push(2)
    s.push(3)
    print(s.pop())
    print(s.pop())
    s.push(4)
    print(s.pop())
    s.push(5)
    print(s.pop())
    print(s.pop())