【剑指 offer】 包含 min 函数的栈

本贴最后更新于 948 天前,其中的信息可能已经东海扬尘

题目

定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 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()
 */

思路

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

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

相关帖子

欢迎来到这里!

我们正在构建一个小众社区,大家在这里相互信任,以平等 • 自由 • 奔放的价值观进行分享交流。最终,希望大家能够找到与自己志同道合的伙伴,共同成长。

注册 关于
请输入回帖内容 ...