剑指 Offer 09. 用两个栈实现队列
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
Solution
class CQueue(object): def __init__(self): self.stackA, self.stackB = [], [] def appendTail(self, value): """ :type value: int :rtype: None """ self.stackA.append(value) def deleteHead(self): """ :rtype: int """ if self.stackB: return self.stackB.pop() if not self.stackA: return -1 while self.stackA: self.stackB.append(self.stackA.pop()) return self.stackB.pop()