剑指 Offer(23) ——包含min函数的栈

332 阅读2分钟

这是我参与8月更文挑战的第23天,活动详情查看:8月更文挑战

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

思路 2.1 解题思路 这道题有两种解题方法,但是本质其实是一样的:就是使用两个栈,一个栈用于存压入的数据,另一个栈用于存储当前栈的最小值,并且能够随着数据的压入和弹出动态的更新当前栈中的最小值。 第一种方式 使用两个栈,数据栈data_stack和最小元素栈min_stack,min_stack栈顶是当前数据栈对应的最小值。 入栈规则:对于要压入的数据value,先将这个数压入数据栈data_stack中。如过这个数value<=min_stack.top(),则将这个数压入min_stack中,否则压入min_stack.top()对应的值到min_stack。 出栈规则:从min_stack中弹出一个数据,从data_stack中弹出数据作为函数的返回结果。 min函数 :返回min_stack.top()对应的值即可。

第二种方式 使用两个栈,数据栈data_stack和最小元素栈min_stack,min_stack栈顶是当前数据栈对应的最小值。 入栈规则:对于要压入的数据value,先将这个数压入数据栈data_stack中。如过这个数value<=min_stack.top(),则将这个数压入min_stack中。 出栈规则:从如果数据栈栈顶的数小于等于min_stack栈顶的数,则从min_stack中中弹出一个数据;从data_stack中弹出数据作为函数的返回结果。 min函数 :返回min_stack.top()对应的值即可。

代码

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

/** 
 * @param {number} x
 * @return {void}
 */
MinStack.prototype.push = function(x) {
    if(!this.stack.length) this.minval = x;
    this.stack.push(x)
    if(x<this.minval) this.minval=x;
    this.minst.push(this.minval)
};

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

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

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

/**
 * 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()
 */