【LeetCode】232. 用栈实现队列（简单）——代码随想录算法训练营Day10

题目链接：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（双端队列）来模拟一个栈，只要是标准的栈操作即可。

示例 1：

输入： ["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?操作）

进阶：

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

文章讲解：代码随想录

视频讲解：栈的基本操作！ | LeetCode：232.用栈实现队列_哔哩哔哩_bilibili

题解1：用两个栈实现队列

思路：一个栈作为入栈，另一个栈作为出栈。当入队操作时，把元素加入到入栈中；出队操作时，如果出栈为空，就将入栈的所有元素加到出栈中，弹出出栈顶部的元素。

var MyQueue = function() {
    this.inStack = [];
    this.outStack = [];
};

/** 
 * @param {number} x
 * @return {void}
 */
MyQueue.prototype.push = function(x) {
    this.inStack.push(x);
};

/**
 * @return {number}
 */
MyQueue.prototype.pop = function() {
    if (this.outStack.length === 0) {
        while (this.inStack.length) {
            this.outStack.push(this.inStack.pop());
        }
    }
    return this.outStack.pop();
};

/**
 * @return {number}
 */
MyQueue.prototype.peek = function() {
    // 复用 pop 方法，先从出栈中弹出，在压入出栈，这个元素即为队首元素
    const num = this.pop();
    this.outStack.push(num);
    return num;
};

/**
 * @return {boolean}
 */
MyQueue.prototype.empty = function() {
    return this.inStack.length === 0 && this.outStack.length === 0;
};

/**
 * Your MyQueue object will be instantiated and called as such:
 * var obj = new MyQueue()
 * obj.push(x)
 * var param_2 = obj.pop()
 * var param_3 = obj.peek()
 * var param_4 = obj.empty()
 */

分析：push 和 empty 的时间复杂度为 O(1)，pop 和 peek 的时间复杂度为 O(n)。空间复杂度为 O(n)。

收获

学习了栈和队列的基本知识，学会了使用两个栈模拟队列。