20-包含min函数的栈

2020-05-08  本文已影响0人  马甲要掉了

题目描述

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。

分析

增加一个辅助栈,每次压入数据栈时,把当前栈里面最小的值压入辅助栈当中。这样辅助栈的栈顶数据一直是数据栈中最小的值。

比如,data中依次入栈, 5, 4, 3, 8, 10, 11, 12, 1
则min依次入栈, 5, 4, 3, 3, 3, 3, 3, 1

代码

let stack = [];
let minStack = [];
let tmp = null;
function push(node)
{
    // write code here
    if(tmp!=null){
        if(tmp>node){
            tmp = node;
        }
        stack.push(node);
        minStack.push(tmp);
    }else{
        tmp = node;
        stack.push(node);
        minStack.push(node);
    }
}
function pop() {
  stack.pop();
  minStack.pop();
}
function top() {
  return stack[stack.length - 1];
}
function min() {
  return minStack[minStack.length - 1];
}
上一篇下一篇

猜你喜欢

热点阅读