代码语言:javascript复制
class Solution(object):
"""
两个队列实现一个栈
两个队列来回倒
"""
def __init__(self):
"""
初始化两个队列
"""
self.queue1 = []
self.queue2 = []
def push(self, element):
# 入栈时 选择一个空的队列
if self.queue1 == []:
self.queue2.append(element)
else:
self.queue1.append(element)
def pop(self):
# 栈为空时 抛异常
if self.queue1 == [] and self.queue2 == []:
raise IndexError("index out of range, stack is empty")
elif self.queue2 == []:
# queue2 为空的时候,把queue1的除了最后一个数全给添加到queue2
for val in range(len(self.queue1) - 1):
self.queue2.append(self.queue1.pop(0))
# 弹出 queue1的元素 只有一个元素
return self.queue1.pop()
else:
# queue1 为空的时候,把queue2的除了最后一个数全给添加到queue1
for val in range(len(self.queue2) -1):
self.queue1.append(self.queue2.pop(0))
# 弹出 queue2的元素 只有一个元素
return self.queue2.pop()