剑指offer 09 用两个栈实现队列
题目描述: 用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail和deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead操作返回 -1 )
算法分析:
队列要求在尾部插入,在头部删除,而栈的插入和删除只能在栈顶(尾部)。但如果考虑使用两个栈,插入的时候在栈A插入,删除时把栈A中的一个个元素取出来放到栈B中,那么B中的栈顶元素即为A中的栈底元素,此时在B中完成删除即可满足队列删除的要求。
【注意】
1.从栈A导入到栈B时用的是pop,故当B为空,不会再一次导入已经出队的数;
2.关键是每当栈B空的时候才从A导入到B,这样就保证了B栈顶是当前队列中最早入队(最旧)的那个数,这是个动态的过程;
3.思维不要局限于基本队列的形状,关键是实现这两个功能。
代码实现:
#python
class CQueue:
def __init__(self):
self.A = []
self.B = []
def appendTail(self, value: int) -> None:
self.A.append(value)
def deleteHead(self) -> int:
if self.B: return self.B.pop()
if not self.A: return -1
while self.A:
self.B.append(self.A.pop())
return self.B.pop()