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 

 

代码：

 1 """
 2 用两个栈实现一个队列
 3 """
 4 class Solution:
 5     def __init__(self):
 6         self.stack1 = []
 7         self.stack2 = []
 8 
 9     def push(self, node):
10         # write code here
11         self.stack1.append(node)
12         
13     def pop(self):
14         # return xx
15         if self.stack1 == []:
16             return None
17         else:
18             for i in range(len(self.stack1)):
19                 self.stack2.append(self.stack1.pop())
20             out = self.stack2.pop()
21             for j in range(len(self.stack2)):
22                 self.stack1.append(self.stack2.pop())
23             return out

 

 

转载于:https://www.cnblogs.com/springionic/p/10576155.html