(※)力扣刷题-栈和队列-用栈实现队列

使用栈实现队列的下列操作:

  • push(x) – 将一个元素放入队列的尾部。
  • pop() – 从队列首部移除元素。
  • peek() – 返回队列首部的元素。
  • empty() – 返回队列是否为空。

说明:
你只能使用标准的栈操作 – 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)。——>判断代码健壮性

思路

这道题考察的就是对栈和队列的掌握程度。
使用栈来模式队列的行为,如果仅仅用一个栈,是一定不行的,所以需要两个栈一个输入栈,一个输出栈,这里要注意输入栈和输出栈的关系。

在push数据的时候,只要数据放进输入栈就好,但在pop的时候,操作就复杂一些,输出栈如果为空,就把进栈数据全部导入进来(注意是全部导入),再从出栈弹出数据,如果输出栈不为空,则直接从出栈弹出数据就可以了。
最后如何判断队列为空呢?如果进栈和出栈都为空的话,说明模拟的队列为空了。

代码

注意一些细节:

  • 判断栈是否为空,才能够进行有效操作
  • pop时候的复杂性
  • peek复用pop后需要把元素加回去
class MyQueue(object):

    def __init__(self):
        """
        in 主要负责push out主要负责pop(用栈实现队列 必然是要用到两个栈)
        """
        self.stack_in = [] # python中没有栈这个数据结构 用列表代替
        self.stack_out = []


    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        # 新元素进来就往 in 中加入 
        # 在push数据的时候,只要数据放进输入栈就好
        self.stack_in.append(x)


    def pop(self):
        """
        :rtype: int
        """
        # 在pop的时候,操作就复杂一些,输出栈如果为空,就把进栈数据全部导入进来(注意是全部导入),
        # 再从出栈弹出数据,如果输出栈不为空,则直接从出栈弹出数据就可以了。
        # 从队列首部移除元素
        if self.empty():
            return None
        if self.stack_out: # stack_out中有元素
            return self.stack_out.pop() # 这里用return 既可以
        else:
            for i in range(len(self.stack_in)):
                self.stack_out.append(self.stack_in.pop()) 
            return self.stack_out.pop()
            

    def peek(self):
        """
        :rtype: int
        """
        ans = self.pop() # 代码 函数复用
        self.stack_out.append(ans) # 因为只是返回元素 所以采用pop删除之后(代码复用)还需要再加回去
        return ans


    def empty(self):
        """
        :rtype: bool
        """
        return not (self.stack_in or self.stack_out)



# 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()

参考:https://www.programmercarl.com/0232.%E7%94%A8%E6%A0%88%E5%AE%9E%E7%8E%B0%E9%98%9F%E5%88%97.html

你可能感兴趣的:(leetcode刷题,leetcode,python,算法,数据结构)