剑指 Offer 09. 用两个栈实现队列(栈模块)

97 阅读2分钟

每日刷题第11天 2021.1.6

用两个栈实现队列

  • 难度:简单
  • 方法:栈

专题:栈

  • 栈(Stack)又名堆栈,它是一种重要的数据结构。
  • 从数据结构角度看,栈也是线性表,其特殊性在于栈的基本操作是线性表操作的子集,它是操作受限的线性表,因此,可称为限定性的数据结构。限定它仅在表尾进行插入或删除操作。表尾称为栈顶,相应地,表头称为栈底。
  • 栈的基本操作除了在栈顶进行插入和删除外,还有栈的初始化,判空以及取栈顶元素等。

题目

  • 用两个栈实现一个队列。
  • 队列的声明如下,请实现它的两个函数 appendTaildeleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )

示例

输入:
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出:[null,null,3,-1]
输入:
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]

提示

  • 1 <= values <= 10000
  • 最多会对 appendTail、deleteHead 进行 10000 次调用

解法

var CQueue = function() {
// 创建2个栈,模拟1个队列
this.stackA = [];
this.stackB = [];
};
/**
* @param {number} value
* @return {void}
*/
CQueue.prototype.appendTail = function(value) {
  this.stackA.push(value);
};
/**
* @return {number}
*/
CQueue.prototype.deleteHead = function() {
// return this.stackA.shift();
// 第二个栈模拟
// 1. 当第二个栈中没有元素的时候,就将第一个栈中的元素倒着存储到第二个栈中
// 2. 当第二个栈中有元素的时候,就将放到第二个栈中,保存在第一个栈中
if (this.stackB.length == 0) {
  let len = this.stackA.length;
  for (let i = 0; i < len; i++) {
    // 将元素倒给栈2中
    this.stackB.push(this.stackA.pop());
  }
}
if (this.stackB.length == 0) {
  return -1;
}
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()
*/

附录

  • 使用数组的shift方法,一个数组就可以实现。
    • 从数组的顶部弹出数