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