每日算法:用两个栈实现队列

[[422522]]

本文转载自微信公众号「三分钟学前端」,作者 sisterAn 。转载本文请联系三分钟学前端公众号。

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

示例 1:

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

示例 2:

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

提示:

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

解题思路:

  • 栈后进先出,队列先进先出
  • 双栈可以实现序列倒置:假设有 stack1=[1, 2, 3] 、 stack2=[] ,如果循环出栈 stack1 并将出栈元素进栈 stack2 ,则循环结束后, stack1=[] 、 stack2=[3, 2, 1] ,即通过 stack2 实现了 stack1 中元素的倒置
  • 当需要删除队首元素时,仅仅需要 stack2 出栈即可;当 stack2 为空时,出队就需要将 stack1 元素倒置倒 stack2 , stack2 再出队即可;如果 stack1 也为空,即队列中没有元素,返回 -1

代码实现:

 
 
 
  1. const CQueue = function() { 
  2.     this.stack1 = [] 
  3.     this.stack2 = [] 
  4. }; 
  5. CQueue.prototype.appendTail = function(value) { 
  6.     this.stack1.push(value) 
  7. }; 
  8. CQueue.prototype.deleteHead = function() { 
  9.     if(this.stack2.length) { 
  10.         return this.stack2.pop() 
  11.     } 
  12.     if(!this.stack1.length) return -1 
  13.     while(this.stack1.length) { 
  14.         this.stack2.push(this.stack1.pop()) 
  15.     } 
  16.     return this.stack2.pop() 
  17. }; 

复杂度分析:

时间复杂度:appendTail 的时间复杂度为O(1),deleteHead 的时间复杂度为 O(n)

 

空间复杂度:O(n)

 

THE END