问题:用两个栈来实现一个队列,完成队列的 Push 和 Pop 操作。
分析:栈的特性是“先进后出”,队列为“先进先出”。
思路:入队:直接把新元素压入stack1即可。
出队:根据队列先进先出的性质,由于先进入队列的元素被压倒stack1的栈底,要想实现先入队列的先出队,需要将stack1中的元素逐个弹出并压入stack2,经过弹出和压入之后最先进入的元素就处于stack2的栈顶,有可以直接弹出。
python实现代码:
class Solution:
def __init__(self):
# 初始化两个空栈
self.stack1 = []
self.stack2 = []
# 实现入队操作
def push(self, node):
# 直接压入 栈1
self.stack1.append(node)
# 实现出队操作
def pop(self):
if len(self.stack2) == 0:
while self.stack1:
self.stack2.append(self.stack1.pop())
return self.stack2.pop()