class CQueue:
def __init__(self):
self.stack_A=[] #栈A
self.stack_B=[] #栈B
#进栈元素存入A中
def appendTail(self, value: int) -> None:
self.stack_A.append(value)
#栈是先进后出,所以需要两个栈来实现队列的功能,先将所有元素存入A中,再一次将A的元素取出放入B
#中,然后再删除B中的元素
#如果A和B都为空,则返回-1
#如果B中有元素先删除B的元素
def deleteHead(self) -> int:
if not self.stack_A and not self.stack_B:
return -1
if self.stack_B:
return self.stack_B.pop(-1)
while self.stack_A:
self.stack_B.append(self.stack_A.pop(-1))
return self.stack_B.pop(-1)
题目:用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。