LeetCode刷题|232用栈实现队列

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

题目描述

LeetCode刷题|232用栈实现队列_第1张图片
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的参考代码,这个是我自己写的,有些混乱,将就着可以看一下

你可能感兴趣的:(算法与数据结构,数据结构,队列,列表,python,栈)