BFE.dev前端刷题#108. 用队列(Queue)实现栈(Stack)

bfe.dev 是一个针对前端的刷题网站，像是前端的LeetCode。该系列文章是我在上面的刷题日记。

题目108

BFE.dev#108 利用队列(Queue)实现栈(Stack)

分析

假设我们有一个队列

[1,2,3,4]

如果我们要pop4的话，因为这是一个队列，我们只能把1 dequeue掉。所以为了要得到4，我们必须要把其余的1,2,3给dequeue掉。dequeue掉的元素因为只能用队列，所以需要用第二个队列来装。比如这样：

[4], [1,2,3]

现在我们可以取得4了。

[ ] , [1,2,3]

这时候我们可以把1，2，3放回去。但是这会让peek()的实现比较麻烦，因为单纯为了peek()也要重复上述操作的话，太浪费了。

为了解决这个问题，我们可以好好利用这两个queue，让其中一个queue始终只有一个元素。比如。

[ ], [ ] 

现在添加1，我们需要enqueue即可。

[1], [ ]

添加2，enqueue过后，我们保持第一个queue只有一个元素，所以把1放入第二个queue。

[1,2], [ ]
[2], [1]

添加3和4的时候，重复相同操作。

[2,3], [1]
[3], [1,2]
[3,4], [1,2]
[4], [1,2,3]

现在我们可以dequeue 4了。

[ ], [1,2,3]

因为第一个queue成了空，我们可以尝试让第二个queue包含只有一个元素。

[1,2], [3]

现在我们可以交换两个queue的地位，这样就变成了和之前完全一样的状态。

[3], [1, 2]

OK，综上所述，我们只需要有两个queue。

• queue1 始终保持只有一个元素（或者初始都为空的状态）
• queueRest 用来存放其余的元素

来，代码写起来

如下代码完全是按照上述想法的复现，就不再说明了。

class Stack {
  constructor() {
    this.queue1 = new Queue()
    this.queueRest = new Queue()
  }
  
  push(element) {
    this.queue1.enqueue(element)
    this._move()
  }
  
  _move() {
    while (this.queue1.size() > 1) {
      this.queueRest.enqueue(this.queue1.dequeue())
    }
  }
  
  peek() { 
    return this.queue1.peek()
  }
  pop() {
    if (this.queue1.size() === 0) {
      return undefined
    }
    
    const element = this.queue1.dequeue()
    
    // swap if the other queue is not empty
    if (this.queueRest.size() > 0) {
      ;[this.queue1, this.queueRest] = [this.queueRest, this.queue1]
      this._move()
    }
    
    return element
  }
  size() { 
    return this.queue1.size() + this.queueRest.size()
  }
}

通过，撒花！

如果你感兴趣可以去BFE.dev 试试 https://bigfrontend.dev/zh/pr...

感谢阅读，希望有所帮助，下次见！