题目描述:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
思路:
使用两个栈实现一个队列的功能,实现的效果图如下:
(图源《算法小抄》)
- 入队:将元素压入s1即可。
- 出队:s2起到一个中转的作用:当s2为空的时候,可以把s1的所有元素去除添加进s2,这时候s2中的元素就是先进先出的顺序了,接着取出元素即可。
代码实现:
var CQueue = function() {
this.s1=[];
this.s2=[];
};
/**
* @param {number} value
* @return {void}
*/
CQueue.prototype.appendTail = function(value) {
this.s1.push(value);
};
/**
* @return {number}
*/
CQueue.prototype.deleteHead = function() {
let {s1,s2}=this;
if(s2.length<=0){
while(s1.length){
s2.push(s1.pop());
}
}
if(s2.length){
return s2.pop();
}else{
return -1;
}
};
/**
* Your CQueue object will be instantiated and called as such:
* var obj = new CQueue()
* obj.appendTail(value)
* var param_2 = obj.deleteHead()
*/
时间复杂度:
- 入队操作:时间复杂度为O(1)。
- 出队操作:最好的时间复杂度为O(1);
最坏的时间复杂度为O(N)。(因为包括while循环,可能需要从s1往s2搬移元素)