LeetCode 题解:232. 用栈实现队列,使用两个栈 入队 - O(1), 出队 - O(n),JavaScript,详细注释

用户头像
Lee Chen
关注
发布于: 2020 年 09 月 10 日
LeetCode题解:232. 用栈实现队列,使用两个栈 入队 - O(1), 出队 - O(n),JavaScript,详细注释

原题链接:https://leetcode-cn.com/problems/implement-queue-using-stacks/



解题思路:



  1. 所有元素都存在s1,同时缓存第一个入队元素为队首。

  2. 当出队时,先将除队首外的元素都缓存到s2,再将s1仅剩的一个元素pop。

  3. pop完成之后,再将s2缓存的元素依次返回s1。



/**
* Initialize your data structure here.
*/
var MyQueue = function () {
this.s1 = []; // 储存队列
this.s2 = []; // pop时临时存储数据
this.top = null; // 储存队首
};
/**
* Push element x to the back of queue.
* @param {number} x
* @return {void}
*/
MyQueue.prototype.push = function (x) {
// 将第一个入队元素存储为队首
if (!this.s1.length) {
this.top = x;
}
// 所有元素正常入栈
this.s1.push(x);
};
/**
* Removes the element from in front of queue and returns that element.
* @return {number}
*/
MyQueue.prototype.pop = function () {
// s1长度为1时,栈会被清空,因此清空队首的值,但此处用例没有校验
if (this.s1.length <= 1) {
this.top = null;
}
// 将s1元素除队首全部都转移到s2
while (this.s1.length > 1) {
this.top = this.s1.pop();
this.s2.push(this.top);
}
// 将队首元素出栈并缓存
const pop = this.s1.pop();
// 让s2元素全部返回s1,保持元素的排列顺序
while (this.s2.length) {
this.s1.push(this.s2.pop());
}
return pop;
};
/**
* Get the front element.
* @return {number}
*/
MyQueue.prototype.peek = function () {
return this.top;
};
/**
* Returns whether the queue is empty.
* @return {boolean}
*/
MyQueue.prototype.empty = function () {
return !this.s1.length;
};



发布于: 2020 年 09 月 10 日 阅读数: 25
用户头像

Lee Chen

关注

还未添加个人签名 2018.08.29 加入

还未添加个人简介

评论

发布
暂无评论
LeetCode题解:232. 用栈实现队列,使用两个栈 入队 - O(1), 出队 - O(n),JavaScript,详细注释