[宝宝也能看懂的活动篇][30-Day LeetCoding Challenge] 第十天

最小栈

Hi 你们好，我是张小猪。欢迎来到『宝宝也能看懂』系列特别篇 - 官方小活动 『30-Day LeetCoding Challenge』。

这里是 4 月 10 号的题，也是题目列表中的第 155 题 -- 『最小栈』

题目描述

设计一个支持 push，pop，top 操做，并能在常数时间内检索到最小元素的栈。

• push(x) —— 将元素 x 推入栈中。
• pop() —— 删除栈顶的元素。
• top() —— 获取栈顶元素。
• getMin() —— 检索栈中的最小元素。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.getMin();   --> 返回 -2.

官方难度

EASY

解决思路

题目要求很简单，就是要封装一个叫作 MinStack 的类，几个操做都是普通的栈操做，不过多了一个返回最小的元素值。因为原生 JS 的 Array 就集合了队列和栈的操做行为，因此咱们能够直接基于它来实现。

直接方案

直接基于原生的 Array 来实现栈的操做，另外咱们用一个变量来记录栈内数据的最小值。这里小猪用一个 _top 指针来标识栈顶的位置，从而避免数据的擦除开销和数组动态缩短的开销，另外在每一次 pop 的时候经过遍历去更新记录的最小值。具体代码以下：

class MinStack {
  constructor() {
    this._top = -1;
    this.data = [];
    this.min = Number.MAX_SAFE_INTEGER;
  }
  push(n) {
    this.data[++this._top] = n;
    n < this.min && (this.min = n);
  }
  pop() {
    this.min = Number.MAX_SAFE_INTEGER;
    for (let i = --this._top; i >= 0; --i) {
      this.data[i] < this.min && (this.min = this.data[i]);
    }
  }
  top() {
    return this.data[this._top];
  }
  getMin() {
    return this.min;
  }
}

优化

上述代码有个问题，pop 操做的时间复杂度实际上是 O(n)，由于须要经过遍从来完成最小值的更新。咱们来尝试把它变成 O(1)。

为了直接更新最小值，那么它可能来自于能够直接计算或者提早算好了存起来。因为数据并无什么数学联系，因此直接计算显然是不现实的，那么咱们就从提早算好来考虑。咱们能够想象一下在插入一个新数据的时候发生的状况：当前栈中的最小值我是知道的；新数据是否最小我也是知道的。那么就至关于插入数据前的最小值和插入数据后的最小值我都知道。那么反过来，它们其实也就是弹出数据前和弹出数据后的最小值。

到此，咱们便获得了实现思路。具体代码以下：

class MinStack {
  constructor() {
    this._top = -1;
    this.data = [];
    this.min = Number.MAX_SAFE_INTEGER;
  }
  push(n) {
    this.data[++this._top] = this.min;
    this.data[++this._top] = n;
    n < this.min && (this.min = n);
  }
  pop() {
    this.min = this.data[--this._top];
    --this._top;
  }
  top() {
    return this.data[this._top];
  }
  getMin() {
    return this.min;
  }
}

总结

优化方式中包含了一点思惟的转换，不过也很是简单，但愿能帮助到有须要的小伙伴。

若是以为不错的话，记得『三连』哦。小猪爱大家哟~

相关连接

• 活动题目列表
• leetcode 题解 repo
• 个人 segmentfault 专栏
• 个人知乎专栏