【leetcode232:用栈实现队列】

leetcode232:用栈实现队列

题目：

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（push、pop、peek、empty）：

实现 MyQueue 类：

• void push(int x) 将元素 x 推到队列的末尾
• int pop() 从队列的开头移除并返回元素
• int peek() 返回队列开头的元素
• boolean empty() 如果队列为空，返回 true ；否则，返回 false

---

思路：

队列先进先出，栈先进后出

在push数据的时候，只要数据放进输入栈；

在pop的时候，

• 如果输出栈为空，就把进栈数据全部导入进来，再从出栈弹出数据，
• 如果输出栈不为空，则直接从出栈弹出数据；

最后判断队列是否为空：如果进栈和出栈都为空的话，说明模拟的队列为空。

在代码实现的时候，peek()的实现，复用了pop()，pop的实现，复用了empty。

---

代码实现（Python）：

class MyQueue:

    def __init__(self):
        """
         初始化为两个列表模拟输入栈和输出栈
        """
        self.stack_in = []
        self.stack_out = []

    def push(self, x: int) -> None:
        """
        模拟入栈
        """
        self.stack_in.append(x)

    def pop(self) -> int:
        """
        先检查模拟的队列（输出栈和输入栈）是否为空
        """
        if self.empty():
            return None

        if self.stack_out:  # 自动进行真值测试，空被视为False
            return self.stack_out.pop()
        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) -> int:
        ans = self.pop()
        self.stack_out.append(ans)
        return ans

    def empty(self) -> bool:
        """
        只要输出栈或者输入栈有元素，说明队列不为空
        为空返回True
        """
        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()

当创建一个 MyQueue 类的实例时（比如使用 q = MyQueue()），q 就是这个实例的变量名（或者说是引用），Python 会自动将实例本身作为第一个参数传递给方法。
而 self 是在这个实例的方法内部用来访问这个实例的属性和其他方法的特殊变量。