剑指-Offer-30-包含min函数的栈

剑指 Offer 30.包含min函数的栈

题目描述:

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

提示:

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

思路

剑指-Offer-30-包含min函数的栈_第1张图片

  1. 如果仅仅是一个栈来得到最小元素的话必须就需要遍历这个栈,那么时间复杂度就会是O(n),因此我们需要一个辅助栈来存储数据栈中的最小元素
  2. 一个数据栈用于实现基本的栈的入栈,出栈等基本操作,一个辅助栈用于存储数据栈的最小元素
  3. 使用辅助栈来存储最小值,每次 push() 时,如果新元素比当前最小值小,则将新元素入辅助栈;每次 pop() 时,如果弹出的元素是当前最小值,则将辅助栈栈顶元素弹出。top() 方法直接返回栈顶元素,min() 方法返回辅助栈的栈顶元素即可。

代码

/**
 * initialize your data structure here.
 */
var MinStack = function() {
    //数据栈 保证完整的栈的基本操作
    this.Stack = []
    //辅助栈 记录数据栈中最小值
    this.MinStack = [Infinity]
    //要保证两栈的一致性
};

/** 
 * @param {number} x
 * @return {void}
 */
MinStack.prototype.push = function(x) {
    //数据栈正常压栈
    this.Stack.push(x)
    //辅助栈判断  辅助栈为空直接压栈 不为空 与辅助栈栈顶元素判断大小 如果小或者相等压入
    if(this.MinStack.length == 0 || x <= this.MinStack[this.MinStack.length-1]){
        this.MinStack.push(x)
    }
};

/**
 * @return {void}
 */
MinStack.prototype.pop = function() {
    //出栈
    //数据栈正常出栈
    //如果辅助栈栈顶元素和出栈元素相等 辅助栈栈顶元素也出栈
    var number = this.Stack.pop()
    if(number == this.MinStack[this.MinStack.length-1]){
        this.MinStack.pop()
    }
};

/**
 * @return {number}
 */
MinStack.prototype.top = function() {
    //返回数据栈栈顶元素
    return this.Stack[this.Stack.length-1]
};

/**
 * @return {number}
 */
MinStack.prototype.min = function() {
    //返回辅助栈栈顶元素
    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()
 */

你可能感兴趣的:(算法,算法,数据结构)