探索字节跳动-最小栈｜刷题打卡

本文正在参与掘金团队号上线活动，点击 查看大厂春招职位

一、题目描述：

设计一个支持 push ，pop ，top 操作，并能在常数时间内检索到最小元素的栈。

push(x) —— 将元素 x 推入栈中。 pop() —— 删除栈顶的元素。 top() —— 获取栈顶元素。 getMin() —— 检索栈中的最小元素。

输入：
["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]

输出：
[null,null,null,null,-3,null,0,-2]

解释：
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.getMin();   --> 返回 -2.

二、思路分析：

要实现栈很简单，用array即可，要实现线性时间取到最小值，第一时间想到的就是最小堆，问题在于这道题又要实现出栈，又要维护堆，使用一个数组一个堆不好实现，因为堆并不方便删除某个指定元素，每次都新建一次堆时间上就没有优势了，于是写了个暴力方案，维护两个列表：stack即为栈,sort为从小到大排序的栈内元素，每次出栈入栈时维护sort

三、AC代码：

var MinStack = function() {
    this.sort = [];
    this.stack = [];
};

MinStack.prototype.push = function(val) {
    this.stack.push(val);
    let index = this.sort.length;
    for(let i in this.sort){
        if (val<this.sort[i]){
            index = i;
            break;
        }
    }
    this.sort = [...this.sort.slice(0,index),val,...this.sort.slice(index,this.sort.length)];
};

MinStack.prototype.pop = function() {
    let popped = this.stack.pop();
    let i = this.sort.indexOf(popped);
    this.sort.splice(i,1);
};

MinStack.prototype.top = function() {
    return this.stack[this.stack.length-1]
};

MinStack.prototype.getMin = function() {
    return this.sort[0]
};

四、总结：

AC之后一看，果然时间惨不忍睹，等翻一翻题解再补时间复杂度低的方案