LeetCode刷题|232用栈实现队列

本文介绍了一种利用两个栈来模拟队列的方法,并提供了一个简单的Python实现案例。该方法中,一个栈用于输入,另一个栈用于输出。当输出栈为空时,所有输入栈中的元素会被转移过来,从而实现了队列的先进先出特性。

今天的每日一题是用栈实现队列,是一个简单题。先来看一下题目描述吧。

题目描述

在这里插入图片描述
LeetCode地址:用栈实现队列

双栈方法

最开始看到这个题我想到的是用一个栈来模拟队列,但是python里面没有栈这个数据结构,只能用列表来模拟栈。但是实现完之后提交虽然通过了,我回来再仔细一看感觉有些不对,题目要求只能用栈的操作,但是我好像还用到了insert函数,我是直接把一个列表当做一个栈来使用了,这样虽然能通过测试,但是是不符合题目要求的。于是看了一下官方题解,原来可以使用两个栈来模拟队列。一个栈作为输入,一个栈作为输出,当输出的栈元素全部都弹出去后,就把输入的栈的与元素都放到输出栈,这样就实现了一个队列。而且除了将输入栈元素放入输出栈,其他操作都是O(1)的时间复杂度。

class MyQueue:

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.instack = []
        self.outstack = []

    def push(self, x: int) -> None:
        """
        Push element x to the back of queue.
        """
        self.instack.append(x)

    def pop(self) -> int:
        """
        Removes the element from in front of queue and returns that element.
        """
        if len(self.outstack)==0:
            self.in_to_out()
        n = len(self.outstack)
        return self.outstack.pop(n-1)

    def peek(self) -> int:
        """
        Get the front element.
        """ 
        if len(self.outstack)==0:
            self.in_to_out()
        n = len(self.outstack)
        return self.outstack[n-1]

    def empty(self) -> bool:
        """
        Returns whether the queue is empty.
        """
        if len(self.instack)==0 and len(self.outstack)==0:
            return True
        else:
            return False
    
    def in_to_out(self) -> None:
        while(self.instack):
            n = len(self.instack)
            self.outstack.append(self.instack[n-1])
            del self.instack[n-1]

# Your MyQueue object will be instantiated and called as such:
# obj = MyQueue()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.peek()
# param_4 = obj.empty()

官方题解没有给出python的参考代码,这个是我自己写的,有些混乱,将就着可以看一下😅

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值