[剑指offer] 两个栈实现队列

题目内容

https://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6?tpId=13&tqId=11158&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

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

---

题目思路

栈的特点是先入后处，后入先出。而队列的特点是先入先出，后入后出。所以我们整两个栈，入栈的时候入第一个。出栈的时候如果另外一个非空，则优先处理另一个。如果空的话，把所有的元素再后入先出的放到另一栈里，负负得正，得到了队列的效果。

---

程序代码

# -*- 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 self.stack2:
            return self.stack2.pop()
        else:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
            return self.stack2.pop()