题目:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
实例:
输入:
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出:[null,null,3,-1]
解题:
博主看到这道题后懵了10min,读不懂题
这道题的意思是 用两个栈来完成进入队列出队列操作,其中出队列,两个栈都为空时返回-1,按照这个思路解题就ok了
下面代码中 A栈的栈顶为队列的尾部,栈底为队列的顶部
class CQueue:
def init(self):
self.A = [] # 尾
self.B = [] # 头
def appendTail(self, value: int) -> None:
while self.A:
self.B.append(self.A.pop())
self.A.append(value)
while self.B:
self.A.append(self.B.pop())
def deleteHead(self) -> int:
if not self.A:
return -1
else: return self.A.pop()
关注博主一起学习吧