剑指offer用两个栈实现队列

题目描述

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

思路

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

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack1=[]
        self.stack2=[]
    def push(self, node):
        # write code here
        if node:
            self.stack1.append(node)
    def pop(self):
        # return xx
        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())