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

用两个栈实现队列

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

分析（来源牛客）

队列的特性是：“先入先出”，栈的特性是：“先入后出”

当我们向模拟的队列插入数 a,b,c 时，假设插入的是 stack1，此时的栈情况为：

栈 stack1：{a,b,c}
栈 stack2：{}
当需要弹出一个数，根据队列的"先进先出"原则，a 先进入，则 a 应该先弹出。但是此时 a 在 stack1 的最下面，将 stack1 中全部元素逐个弹出压入 stack2，现在可以正确的从 stack2 中弹出 a，此时的栈情况为：

栈 stack1：{}
栈 stack2：{c,b}
继续弹出一个数，b 比 c 先进入"队列"，b 弹出，注意此时 b 在 stack2 的栈顶，可直接弹出，此时的栈情况为：

栈 stack1：{}
栈 stack2：{c}
此时向模拟队列插入一个数 d，还是插入 stack1，此时的栈情况为：

栈 stack1：{d}
栈 stack2：{c}
弹出一个数，c 比 d 先进入，c 弹出，注意此时 c 在 stack2 的栈顶，可直接弹出，此时的栈情况为：

栈 stack1：{d}
栈 stack2：{c}

总结：

当插入时，直接插入 stack1；

当弹出时，当 stack2 不为空，弹出 stack2 栈顶元素，如果 stack2 为空，将 stack1 中的全部数逐个出栈入栈 stack2，再弹出 stack2 栈顶元素；

class Solution:
    def __init__(self):
        self.stackpush = []   #引入一个压栈
        self.outputstack = []   #引入一个出栈
    def push(self, node):
        # write code here
        self.stackpush.append(node) #插入元素
    def pop(self):
        # return xx
        #先判断出栈是否为空，如果不是直接弹出
        if len(self.outputstack) == 0:
            while self.stackpush: #当压栈不为空的时候，把每个元素弹出到出栈中
                self.outputstack.append(self.stackpush.pop())
        return self.outputstack.pop() #弹出

复杂度

push时间复杂度：O(1)
pop空间复杂度：O(1)