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

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

题目描述：

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

示例1：

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

示例2：

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

提示：

1. 1 <= values <= 10000
2. 最多会对appendTail、deleteHead进行 10000 次调用

---

思路

1. 两个栈实现队列，栈A存储数据，栈B用于将栈A中的元素倒序，再出栈，即将队首删除

2. 例如：栈A：[1,2,3] 想要删队首删A，那么将栈A依次出栈，栈B依次入栈，栈B：[3,2,1] ，这时候再将栈B栈顶元素出栈，即元素为1，删除了队首元素

3. 算法思想有了，那么就需要讨论特定的不同情况了

   （1）如果栈B元素为空，则需要将栈A的元素依次出栈，压入栈B，这时候再将栈B的栈顶元素出栈

   （2）当栈B元素为空的同时，栈A元素也为空（后续没有压入栈A的操作即没有入队的操作），这时候队列为空返回-1

   （3）当栈B有元素，栈B的栈顶元素直接出栈

代码

var CQueue = function() {
    //初始化两个栈
    this.stackA = []
    this.stackB = []
};

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

/**
 * @return {number}
 */
CQueue.prototype.deleteHead = function() {
    //判断栈A和栈B是否同时为空 同时为空返回-1
    if(this.stackA.length === 0 && this.stackB.length === 0){
        return -1
    }
    //如果B为空 将A中的元素依次出栈 在B中依次入栈
    if(this.stackB.length === 0){
        while(this.stackA.length){
            //A出栈 B入栈
            this.stackB.push(this.stackA.pop())
        }
        //栈B栈顶元素出栈
        return this.stackB.pop()
    }else{
        //栈B不为空 直接栈B的栈顶元素直接出栈
        return this.stackB.pop()
    }
};

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