面试题30. 包含min函数的栈|刷题打卡

271 阅读2分钟

本文正在参与掘金团队号上线活动,点击 查看详情

一、题目描述:

定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。

示例:

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

注意:

各函数的调用总次数不超过 20000 次。

二、思路分析:

实现一个栈包含栈的push、pop、min、top。从此可以看出需要创建一个MinStack类,类中包含以上四个方法。

可以使用两个栈实现min函数栈。一个存放正常的元素栈(element),一个存放最小元素的栈(smallest)。 最小元素栈从大到小排列。

咱们先挑简单的实现。

top:

返回元素栈顶即可。(this.element[this.element.length - 1])

min:

min方法返回的是当前元素栈中最小的值。直接返回最小栈中的栈顶即可。(this.smallest[this.smallest.length - 1])

push:

对于入栈,可以使用一个元素栈存放正常的元素顺序。

方法一:

在最小栈中,

  • 最小栈为空。直接将当前元素push到最小栈中;
  • 不为空,判断存放的当前值与当前最小栈的栈顶元素:小于等于,执行压栈操作。

例:

minStack.push(-2);
minStack.push(0);
minStack.push(-3);

image.png

方法二: 最小栈的长度和元素栈一样。

  • 最小栈为空。直接将当前元素push到最小栈中;
  • 最小栈不为空。 当前元素小于等于最小栈栈顶元素,将当前元素压入最小栈;大于,将当前最小栈栈顶元素再一次入栈。

例:

minStack.push(-2);
minStack.push(0);
minStack.push(-3);

image.png

pop:

方法一:

采用push方法一后,在考虑出栈的时候,需要判断当前最小栈栈顶是否等于当前元素的栈顶元素。 如果相等,在元素栈出栈时,最小栈也需要执行相应出栈操作。 例:

minStack.pop();
minStack.pop();

image.png

方法二: 沿用push方法二:元素出栈时,因为最小栈和元素栈长度相同,所以最小栈也执行相应出栈操作即可。

例:

minStack.pop();
minStack.pop();

image.png

三、AC 代码:

// 法一
class MinStack {
    element: number[];
    smallest: number[];
    constructor() {
        this.element = [];
        this.smallest = [];
    }
    push(x: number): void {
        this.element.push(x);
        if (!this.smallest.length) {
            this.smallest.push(x);
        } else {
            if (this.smallest[this.smallest.length - 1] >= x) {
                this.smallest.push(x);
            }
        }
    }

    pop(): void {
        this.smallest[this.smallest.length - 1] === this.element.pop() && this.smallest.pop();
    }

    top(): number {
        return this.element[this.element.length - 1];
    }

    min(): number {
        return this.smallest[this.smallest.length - 1];
    }
}
// 法二
push(num: number) {
    this.element.push(num);
    if (!this.smallest.length) {
      this.smallest.push(num);
    } else if (this.smallest[this.smallest.length - 1] >= num) {
      this.smallest.push(num);
    } else {
      this.smallest.push(this.smallest[this.smallest.length - 1]);
    }
  }
pop() {
    this.element.pop();
    this.smallest.pop();
  }

四、总结:

考察对栈的基本实现功能。此题属于简单题,对于栈的原理理解,基本可以做出来。