【小白刷剑指offer】第五题用两个栈实现队列

最新推荐文章于 2024-04-30 07:56:37 发布

原创最新推荐文章于 2024-04-30 07:56:37 发布 · 255 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#用两个栈实现队列 #剑指offer

力扣题解专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法，详细解释了如何通过栈的特性完成队列的基本操作：push和pop。文章的核心在于巧妙利用两个栈的不同角色：一个栈用于入队(push)，另一个栈用于出队(pop)，并通过适当的转换机制确保队列的先进先出(FIFO)原则得以实现。

5 用两个栈实现队列

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

class Solution:
    def __init__(self):
        self.stack1=[]
        self.stack2=[]

    def push(self, node):
        # write code here
        self.stack1.append(node)
        print('push',node,'|',self.stack1,self.stack2)

    def pop(self):
        if self.stack2:
            t=self.stack2.pop()
        elif self.stack1 :
            for i in range(len(self.stack1)):
                self.stack2.append(self.stack1.pop())
            t=self.stack2.pop()
        print('pop',t,'|',self.stack1,self.stack2)
        return t

task =Solution()
task.stack1=[]
task.stack2=[]
task.push([1])
task.push([2])
task.pop()
task.pop()
task.push([3])
task.push([4])
task.push([5])
task.pop()
task.pop()
task.pop()

''' 注意
stack1用于push，stack2用于pop
self.stack2.append(self.stack1.pop()) 这里需要用pop，把 stack1 清空
数据结构 self.stack1、self.stack2
'''