LeetCode:剑指offer 09:用两个栈实现队列

erzhuerzhu

墨衍会员 · AI 创作全网分发码龄8年

关注

原创最新推荐文章于 2026-06-23 11:24:30 发布 308 阅读 AI 写同款· GEO 优化 ›

vcs 覆盖率收集1——三种覆盖率类型、covergroup、bins、数据采样、覆盖率选项介绍各种覆盖率，以及如何编写覆盖率收集的代码阅读详情

problem

LeetCode 剑指offer 09:用两个栈实现队列

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

solution

方法一

思路过程

明白栈和队列两种数据结构，栈先进后出，队列先进先出
每一次添加元素，直接放在一个栈的后面
每一次删除元素，把栈的元素pop放到另外一个栈中append，删除中转的栈的栈顶元素，再放回原来的栈中
也可以每次添加元素，用另外一个栈配合放到栈底，每次删除元素就直接出栈，就是看你在来回换的时候放到什么函数中去实现

代码

python

class CQueue:

    def __init__(self):
        # temp作为临时调用的栈
        self.temp = []
        self.queue = []

    def appendTail(self, value: int) -> None:
        self.queue.append(value)

    def deleteHead(self) -> int:
        # 这里真的是傻了，for在in i 循环里面是变化的，就没了
        if self.queue:
            length = len(self.queue)
            for i in range(length):
                self.temp.append(self.queue.pop())
            res = self.temp.pop()
            for i in range(length - 1):
                self.queue.append(self.temp.pop())
            return res
        return -1

优化:

class CQueue:

    def __init__(self):
        # temp作为临时调用的栈
        self.temp = []
        self.queue = []

    def appendTail(self, value: int) -> None:
        self.queue.append(value)

    def deleteHead(self) -> int:
        if self.queue:
            while self.queue:
                self.temp.append(self.queue.pop())
            res = self.temp.pop()
            while self.temp:
                self.queue.append(self.temp.pop())
            return res
        return -1

方法二

思路过程

参考这篇文章☞[双栈，一个负责入，一个负责出]，写得很好(https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/solution/xiong-mao-shua-ti-python3-shuang-zhan-yi-ge-fu-ze-/)
- 两个栈instack 和 outstack，一个用来添加元素，一个用来删除元素
- 调用添加元素的函数时，直接放到instack 栈顶
- 调用减少元素的函数时，查看outstack 是否有元素，如果有，就直接pop，没有就查看instack
- instack有的话，把元素放入outstack，否则返回-1，
这样就避免了每次增加或是删除元素的时候，都要把元素来回倒腾一次，做到了较好的优化，跑起来果然有很大的提升

代码

python

class CQueue:

    def __init__(self):
        self.instack=[]
        self.outstack=[]

    def appendTail(self, value: int) -> None:
        self.instack.append(value)

    def deleteHead(self) -> int:
        if self.outstack:
            return self.outstack.pop()
        if self.instack:
            while self.instack:
                self.outstack.append(self.instack.pop())
            return self.outstack.pop()
        return -1