题目:用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
代码():
# -*- coding:utf-8 -*-
class Solution:
def __init__(self):
self.stack_A = []
self.stack_B = []
def push(self, node):
# write code here
self.stack_A.append(node)
def pop(self):
if (not self.stack_B) and (not self.stack_A):
return None
elif self.stack_B:
return self.stack_B.pop()
else:
while self.stack_A:
self.stack_B.append(self.stack_A.pop())
return self.stack_B.pop()
解题思路:
栈是先进后厨FILO
队列是先进先出FIFO
这里有两个栈 A B
那么想法就是 先把东西放进A 然后如果B为空就把A里面的放进B。
例如:原数列为 1 2 3 4 5 进入A后变为 5 4 3 2 1 再放进B就变成 1 2 3 4 5
如果B不为空,则先把B里面的先输出,输出完毕后再填补A的数据进去