【LeetCode】算法 —— 232. 用栈实现队列【队列】

版权声明：本文为博主原创文章，未经博主允许不得转载。

---

---

描述

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

实现 MyQueue 类

void push(int x) 将元素 x 推到队列的末尾

int pop() 从队列的开头移除并返回元素

int peek() 返回队列开头的元素

boolean empty() 如果队列为空，返回 true ；否则，返回 false

说明

你 只能 使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。

你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可。

示例

输入：
["MyQueue", "push", "push", "peek", "pop", "empty"]
[[], [1], [2], [], [], []]
输出：
[null, null, null, 1, 1, false]


解释：
MyQueue myQueue = new MyQueue();
myQueue.push(1); // queue is: [1]
myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue)
myQueue.peek(); // return 1
myQueue.pop(); // return 1, queue is [2]
myQueue.empty(); // return false

提示

• 1 <= x <= 9
• 最多调用 100 次 push、pop、peek 和 empty
• 假设所有操作都是有效的 （例如，一个空的队列不会调用 pop 或者 peek 操作）
• 原题：https://leetcode.cn/problems/implement-queue-using-stacks/

进阶

• 你能否实现每个操作均摊时间复杂度为 O(1) 的队列？换句话说，执行 n 个操作的总时间复杂度为 O(n) ，即使其中一个操作可能花费较长时间。

思路分析

分析：准备 2 个栈： inStack、 outStack

• 入队时， push 到 inStack 中
  
• 出队时，情况一：如果 outStack 为空，将 inStack 所有元素逐一弹出， push 到 outStack， outStack 再弹出栈顶元素【11出队】
  
• 出队时，情况二：如果 outStack 不为空， outStack 弹出栈顶元素【11】
  

实例：假设如下操作： 11入队、 22入队、出队（11）、 33入队、出队（22——>33）

• 第一步：准备 2个空的 Stack
  
• 第二步：11入队、22入队
  
• 第三步：11 出队：情况一：如果 outStack 为空，将 inStack 所有元素逐一弹出， push 到 outStack， outStack 再弹出栈顶元素【11弹出】
  
  
• 第四步：33入队
  
• 第五步：22 出队：情况二：如果 outStack 不为空， outStack 弹出栈顶元素【22】
  
• 第六步：33 出队：情况一

class MyQueue:

    def __init__(self):
        """
        Initializer your data structure here.
        """
        self.s1 = []
        self.s2 = []
        self.front = None


    # 入队
    def push(self, x: int) -> None:
        """
        Push element x to the back of queue.
        """
        if not self.s1: self.front = x
        self.s1.append(x)

    # 出队
    def pop(self) -> int:
        """
        Removes the element from in front of queue and return that element.
        """
        if not self.s2:
            while self.s1:
                self.s2.append(self.s1.pop())
            self.front = None
        return self.s2.pop()

    # 获取对头元素
    def peek(self) -> int:
        """
        Get the front element.
        """
        if self.s2:
            return self.s2[-1]
        return self.front

    # 是否为空
    def empty(self) -> bool:
        """
        Returns whether the queue is empty.
        """
        if not self.s1 and not self.s2:
            return True
        return False



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

---

版权声明：本文为博主原创文章，未经博主允许不得转载。