剑指Offer(Python多种思路实现):用两个栈实现队列
面试9题:
题目:用两个栈实现队列
题目描述:用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
解题思路一:
class Solution:
def __init__(self):
self.stackA=[]
self.stackB=[]
def push(self, node):
self.stackA.append(node)
def pop(self):
if self.stackB:
return self.stackB.pop()
elif not self.stackA:
return None
else:
while self.stackA:
self.stackB.append(self.stackA.pop())
return self.stackB.pop()
解题思路二:两个队列实现栈
# 两个队列实现栈
class Solution:
def __init__(self):
self.queueA=[]
self.queueB=[]
def push(self, node):
self.queueA.insert(0, node)
def pop(self):
if not self.queueA:
return None
while len(self.queueA) != 1:
self.queueB.insert(0, self.queueA.pop())
self.queueA, self.queueB = self.queueB, self.queueA
return self.queueB.pop()