剑指 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]
复制代码

提示:
1 <= values <= 10000
最多会对 appendTail、deleteHead 进行 10000 次调用

思路:
栈是先入后出,队列是先入先出。
向队列尾部插入整数时,相当于入栈。
在队列头部删除整数时,可利用栈先入后出的特性,把栈1的数据依次放入栈2,此时删除栈顶元素即是删除队列头部整数。

var CQueue = function(){
    this.stack1 = [];
    this.stack2 = [];
}

CQueue.prototype.appendTail = function(value){
    this.stack1.push(value);
}

CQueue.prototype.deleteHead = function(){
    if(this.stack2.length === 0){
        while(this.stack1.length > 0){
            this.stack2.push(this.stack1.pop());
        }
    }
    return this.stack2.pop() || -1;
}
复制代码

2.png

© 版权声明
THE END
喜欢就支持一下吧
点赞0 分享