（※）力扣刷题-栈和队列-用栈实现队列

news2024/9/24 17:19:29

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

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