题目
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解
思路
用两个栈模拟队列,队列是先进先出,用A栈进数据,B栈出数据。
- 向队列尾部添加数据时,直接push到A栈,
-
需要取数据时,先检查B栈有没有数据(第一次取数据时B栈是空的)。 如果没有数据,需要对A栈进行pop,pop返回的数据push到B栈里面,直到把A栈清空。再从B栈pop数据。
代码
var CQueue = function() {
this.stackA=[];
this.stackB=[];
};
/**
* @param {number} value
* @return {void}
*/
CQueue.prototype.appendTail = function(value) {
this.stackA.push(value);
};
/**
* @return {number}
*/
CQueue.prototype.deleteHead = function() {
if(!this.stackB.length){
if(!this.stackA.length)return -1;
else
while(this.stackA.length)
this.stackB.push(this.stackA.pop());
}
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()
*/