Python两个栈实现一个队列

牛客网原题：

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

 

实现这个算法的方式有很多种，这里就写一种比较简单易懂的；虽然可能算法和效率上不太出色，当大多数人都应该是一看就能懂；如果要想寻找效率高算法佳的方法，网上搜索一下会找到很多；

 

思路：

1. 将stack1作为存储空间，将stack2作为临时缓冲区；也就是stack2辅助stack1做出队与入队操作；

2. 入队时，直接将元素压入stack1即可；

3. 出队时，将stack1中的元素依次出栈压入stack2中，再将stack2的栈顶元素弹出，最后将stack2中的元素再依次出栈压入stack1中，从而实现元素的出队，并返回弹出的元素；

4. 更多思路，访问博客地址：https://www.cnblogs.com/tracyhan/p/5490775.html 

 

代码：

"""
用两个栈实现一个队列
"""
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.stack1 == []:
            return None
        else:
            for i in range(len(self.stack1)):
                self.stack2.append(self.stack1.pop())
            out = self.stack2.pop()
            for j in range(len(self.stack2)):
                self.stack1.append(self.stack2.pop())
            return out