用栈实现队列
题目
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push
、pop
、peek
、empty
):
实现 MyQueue
类:
void push(int x)
将元素 x 推到队列的末尾int pop()
从队列的开头移除并返回元素int peek()
返回队列开头的元素boolean empty()
如果队列为空,返回true
;否则,返回false
说明:
- 你 只能 使用标准的栈操作 —— 也就是只有
push to top
,peek/pop from top
,size
, 和is empty
操作是合法的。 - 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
示例 1:
输入:
["MyQueue", "push", "push", "peek", "pop", "empty"]
[[], [1], [2], [], [], []]
输出:
[null, null, null, 1, 1, false]
解释:
MyQueue myQueue = new MyQueue();
myQueue.push(1); // queue is: [1]
myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue)
myQueue.peek(); // return 1
myQueue.pop(); // return 1, queue is [2]
myQueue.empty(); // return false
思路
双栈实现队列核心是利用两个栈分别存入入栈数据列,出栈数据列。
- 设计入队数据,即将push进入队列的数据存入储存栈(inStack)。
- 当读取到pop出队命令时,将储存栈数据一一pop并存入读取站(outStack),这样在读取栈最后的数据就是储存栈中最开始的数据,这时候pop读取栈的数据,这个数据也就是最开始存入的数据,这样我们就到达了先进先出的效果。
- 在读取过程中又一次存入数据,这时候我们也不用管,一直到读取栈的数据全部被读取完毕,为空时,我们再一次将存储栈中的数据倒置入读取栈即可。
复杂度
- 时间复杂度:
push和empty为 O ( 1 ) O(1) O(1)pop和peek为均摊 O ( 1 ) O(1) O(1)。对于每个元素,至多入栈和出栈各两次,故均摊复杂度为 O ( 1 ) O(1) O(1)。
- 空间复杂度:
O ( n ) O(n) O(n)
代码
var MyQueue = function() {
this.inStack=[],
this.outStack=[]
};
/**
* @param {number} x
* @return {void}
*/
//<!-- 向队列存入数据,即向储存栈最后端存入数据 -->
MyQueue.prototype.push = function(x) {
this.inStack.push(x);
};
/**
* @return {number}
*/
//<!-- 删除队列第一个元素,即删除最先入栈的元素,即若读取栈为空,则该数据在储存栈最低端,我们需要先把储存栈中元素全部移入读取栈,然后读取栈顶元素即最后的那一个元素 -->
MyQueue.prototype.pop = function() {
if(this.outStack.length!==0) return this.outStack.pop();
else {
this.move();
return this.outStack.pop();
}
};
/**
* @return {number}
*/
//<!--读取队列第一个元素,也就是读取读取栈的最后端(顶端)的元素 -->
MyQueue.prototype.peek = function() {
if(this.outStack.length!==0) return this.outStack[this.outStack.length-1];
else {
this.move();
// console.log(this.outStack.length-1)
return this.outStack[this.outStack.length-1];
}
};
/**
* @return {boolean}
*/
//<!-- 队列判空只有当储存栈和读取站两个栈同时位空时,队列才算为空 -->
MyQueue.prototype.empty = function() {
if(this.inStack.length===0&&this.outStack.length===0) return true;
else return false;
};
MyQueue.prototype.move = function(){//将储存栈中的数据倒置入读取栈
while(this.inStack.length!==0){
let x = this.inStack.pop();
this.outStack.push(x);
}
}
/**
* Your MyQueue object will be instantiated and called as such:
* var obj = new MyQueue()
* obj.push(x)
* var param_2 = obj.pop()
* var param_3 = obj.peek()
* var param_4 = obj.empty()
*/