剑指 Offer 09. 用两个栈实现队列

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

示例 1：

输入：
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出：[null,null,3,-1]

示例 2：

输入：
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出：[null,-1,null,null,5,2]

题解：

var CQueue = function () {
    this.stack = [];
    this.outStack = []
};

/** 
 * @param {number} value
 * @return {void}
 */
CQueue.prototype.appendTail = function (value) {
    this.stack.push(value)
};

/**
 * @return {number}
 */
CQueue.prototype.deleteHead = function () {
    // 方法一：数组的shift方法
    if (this.stack.length) {
        let res = this.stack.shift()
        return res
    } else {
        return -1
    }
    // 方法二：利用双栈结构
    // 当辅助栈内无数据时进入
    if (!this.outStack.length) {
        // 当主栈内无数据时返回 -1
        if (!this.stack.length) {
            return -1
        }
        // 当主栈内有数据时，把主栈内数据栈顶元素
        // 删除，放入辅助栈内 => 辅助栈内数据为主栈内倒叙的数据
        while (this.stack.length) {
            this.outStack.push(this.stack.pop())
        }
    }
    return this.outStack.pop()
};

/**
 * Your CQueue object will be instantiated and called as such:
 * var obj = new CQueue()
 * obj.appendTail(value)
 * var param_2 = obj.deleteHead()
 */

来源：力扣（LeetCode）

链接：leetcode.cn/problems/yo…

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。