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

题目地址：包含min函数的栈

 

题目描述                                   

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O(1)）。 

    

解法分析                                   

一般来说返回最小值需要遍历整个栈，时间复杂度为O(n)，不满足题目要求的O(1)，因此我们需要考虑使用其他方法，加个辅助栈用于存放最小元素，每次有元素进栈就与辅助栈元素作对比。

 

代码                                         

var minStack = [];
var dataStack = [];
function push(node)
{
    // write code here
    dataStack.push(node);
    if(minStack.length == 0 || node < minStack[minStack.length-1]){
        minStack.push(node);
    }else{
        minStack.push(minStack[minStack.length-1]);
    }
}
function pop()
{
    // write code here
    if(minStack.length > 0 && dataStack.length >0){
        minStack.pop();
        dataStack.pop();
    }
    
}
function top()
{
    // write code here
    if(dataStack.length > 0) return dataStack[dataStack.length-1];
}
function min()
{
    // write code here
    if(dataStack.length > 0 && minStack.length > 0) return minStack[minStack.length-1];
}

 

执行结果