【剑指offer】Q9_用两个栈实现队列

最新推荐文章于 2025-12-03 12:28:21 发布

原创最新推荐文章于 2025-12-03 12:28:21 发布 · 229 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#栈 #队列 #python

剑指offer 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法，通过在两个栈之间转移元素，可以有效地进行队列的Push和Pop操作。文章提供了详细的Python代码实现，并展示了如何在队列中插入和删除整数类型的元素。

题目

用两个栈来实现一个队列，完成队列的Push和Pop操作。队列中的元素为int类型。

解答

两个栈来回倒就能访问栈底了，没啥说的

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stark1 = []  # 用list模拟栈 在后续代码中不得违反栈的操作规则
        self.stark2 = []  # 即只能对其 append 或 pop

    def push(self, node):
        # 搬移元素
        for _ in range(len(self.stark2)):
            self.stark1.append(self.stark2.pop())
        # 插入元素
        self.stark1.append(node)

    def pop(self):
        # 搬移元素
        for _ in range(len(self.stark1)):
            self.stark2.append(self.stark1.pop())
        # 弹出元素
        return self.stark2.pop()


s = Solution()
# ["PSH1","PSH2","PSH3","POP","POP","PSH4","POP","PSH5","POP","POP"]
s.push(1)
s.push(2)
s.push(3)
print(s.pop())
print(s.pop())
s.push(4)
print(s.pop())
s.push(5)
print(s.pop())
print(s.pop())
# 对应输出应该为: 1,2,3,4,5