leetcode--用栈实现队列

最新推荐文章于 2025-03-17 20:58:54 发布

原创最新推荐文章于 2025-03-17 20:58:54 发布 · 165 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LeetCode 专栏收录该内容

73 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法，通过将一个栈的元素倒置到另一个栈，可以实现在队列头部删除元素的功能。当队列头部需要删除元素时，检查辅助栈是否为空，若为空则将主要栈的所有元素依次弹出并压入辅助栈，最后从辅助栈弹出顶部元素即为队列头部元素。

题目来源：链接

题目描述：

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

实现代码：

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()

解题思路：

因为要实现的队列的加入队尾和删除队首，所以就将A的pop元素加入到B中，B的pop元素就是原队列的deleteHead，当然也可以用双端队列