剑指Offer--5.用两个栈实现队列

题目描述：

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

python代码：

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack1 = []
        self.stack2 = []
    def push(self, node):
        # write code here
        self.stack1.append(node)
    def pop(self):
        # return xx
        if len(self.stack2) != 0:
            return self.stack2.pop()
        else:
            while len(self.stack1):
                self.stack2.append(self.stack1.pop())
            if len(self.stack2) > 0:
                return self.stack2.pop()
            else:
                return None

解析：

队列是先进先出，栈是先进后出，正好两个栈的组合可以实现先进先出。入队时，直接push进stack1，而出队时，要先判断stack2中是否有元素，如果有则从stack2中弹出，如果没有，需要把stack1中的所有元素压入stack2中，正好最先进入的stack1的能够变成stack2的栈顶元素。正是因为出队时将stack1中的所有元素都压入了stack2，因此出队时一定要先考虑stack2中的栈顶元素，只有stack2中没有元素，才能从stack1中弹出并压入stack2才能保持队列的顺序。