一、题目描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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]
二、解题方法
解题思路
-
从队列的声明中可以看出,一个队列包含了两个栈,利用栈的先进后出,完成队列的先进先出,定义两个栈分别为 stack1和stack2
-
插入元素
首先插入一个元素a,插入到stack1中,此时stack2为空,再压入两个元素b,c,,此时stack1中有三个元素,顺序为{c,b,a}而stack2中依然为空 -
删除元素
按照队列的先进先出原则,因为a最先插入,即应该按照插入的顺序删除,先删除a,再依次删除b,c 因为stack2中一直为空,我们可以利用stack1,把所有元素依次弹出并插入stack2中,此时stack2中的元素顺序为{a,b,c},即a位于栈顶,然后再依次弹出,就可以满足队列的要求。
算法流程: -
首先创建两个数组,分别为stack1,stack2
-
appendTail插入方法,利用push方法插入到stack1中
-
deleteHead弹出方法,首先判断stack2是否为空,如果不为空,弹出stack2栈顶元素即可,如果为空,依次把stack1 中的元素弹出到stack2中
-
代码如下
var CQueue = function() {
this.stack1=[];
this.stack2=[]
};
/**
* @param {number} value
* @return {void}
*/
CQueue.prototype.appendTail = function(value) {
this.stack1.push(value)
};
/**
* @return {number}
*/
CQueue.prototype.deleteHead = function() {
if(this.stack2.length===0){
while(this.stack1.length>0){
this.stack2.push(this.stack1.pop())
}
}
if(this.stack2.length>0){
return this.stack2.pop()
}else{
return -1
}
};