[路飞]_leetcode面试题 03.04. 化栈为队

151 阅读2分钟

题目描述

面试题 03.04. 化栈为队

实现一个MyQueue类,该类用两个栈来实现一个队列。

示例

MyQueue queue = new MyQueue();

queue.push(1);
queue.push(2);
queue.peek();  // 返回 1
queue.pop();   // 返回 1
queue.empty(); // 返回 false

说明

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

解题思路分析

image.png

image.png

JavaScript实现

// 队列元素是头部出队,尾部入队的,先入先出的一种结构;
// 栈 是先入后出的结构
// 用两个栈实现一个队列,也就是说具体实现是使用的栈,但是外在表现同队列一样,即先入先出
// 用栈模拟队列,入队与入栈都是从尾部进入;区别:出栈从尾部,出队列是从头部
// 定义两个栈,s1,s2;  元素入栈s2, 出栈时,把元素逆序加入s1, 这时两个栈元素顺序是相反的,然后让s1尾部元素出栈。
// 具体实现步骤:
// js中使用数组模拟
// 1. 定义两个数组 s1(出栈操作),s2(入栈操作)
// 2. 当s1为空的话(初始或者有元素但已经全部出栈了),如果s2有元素,把s2依次加入s1

/**
 * Initialize your data structure here.
 */
let s1, s2;
var MyQueue = function() {
    s1 = [];
    s2 = [];

};
// 将s2的元素移动到s1,基于原序列的一个逆序的操作
function transfer() {
    if(s1.length) return;
    // s1空的时候,把s2的全部元素移动到s1
    while(s2.length) {
        let last = s2.length - 1
        s1.push(s2[last])
        s2.pop()
    }
}
/**
 * Push element x to the back of queue. 
 * @param {number} x
 * @return {void}
 */
MyQueue.prototype.push = function(x) {
    // 入队操作
    s2.push(x);
};

/**
 * Removes the element from in front of queue and returns that element.
 * @return {number}
 */
MyQueue.prototype.pop = function() {
    // 出队操作, 
    transfer()
    let last = s1.length - 1
    let ret = s1[last];//尾部元素
    s1.pop()//出队
    return ret;
};

/**
 * Get the front element.
 * @return {number}
 */
MyQueue.prototype.peek = function() {
    // 查看队首元素---- 数组s1 的最后一个元素
    transfer()
    return s1[s1.length - 1]
};

/**
 * Returns whether the queue is empty.
 * @return {boolean}
 */
MyQueue.prototype.empty = function() {
    // 判空,s1与s2都为空
    return !s1.length && !s2.length
};

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