limanting 的个人博客

题目

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

我的答案

/**
 * initialize your data structure here.
 */
var MinStack = function () {
    this.stack = [];
    this.minStack = [];
};

/** 
 * @param {number} x
 * @return {void}
 */
MinStack.prototype.push = function (x) {
    this.stack.push(x);
    if (!this.minStack.length || this.minStack[this.minStack.length - 1] >= x) {
        this.minStack.push(x);
    }
};

/**
 * @return {void}
 */
MinStack.prototype.pop = function () {
    const item = this.stack.pop();
    if (item === this.minStack[this.minStack.length - 1]) {
        this.minStack.pop()
    }
    return item;
};

/**
 * @return {number}
 */
MinStack.prototype.top = function () {
    return this.stack[this.stack.length - 1];
};

/**
 * @return {number}
 */
MinStack.prototype.min = function () {
    if (this.minStack.length) {
        return this.minStack[this.minStack.length - 1];
    }
};

/**
 * Your MinStack object will be instantiated and called as such:
 * var obj = new MinStack()
 * obj.push(x)
 * obj.pop()
 * var param_3 = obj.top()
 * var param_4 = obj.min()
 */

思路

因为要求复杂度都是一，所以算法其实有点难。这里是利用了堆栈的特点，先进后出，只需要控制后出的元素一定比当前的最小值小，所以还需要一个堆栈存储最小值。

在将新的元素推入堆栈中时，对比存储最小值堆栈栈顶的元素。只有新的元素比存储最小值栈顶元素小或相等时，再将该元素推入存储最小值的栈中。