题解 | #包含min函数的栈#

包含min函数的栈

http://www.nowcoder.com/practice/4c776177d2c04c2494f2555c9fcc1e49

方法一:辅助栈
核心思想
普通栈的 函数的复杂度为,满足题目要求。但获取栈最小值的 函数需要遍历整个栈,复杂度为,所以本题的主要目的就在于优化对栈的最小值的获取,这可以通过使用一个具有单调性的辅助栈来实现。
具体思路:
数据栈:数据栈用于存储所有元素,保证函数的正常逻辑。
辅助栈:辅助栈中存储数据栈中所有降序的元素。栈具有先进后出的特点,所以一个值压入栈中后,只要未被弹出,所有在它之后进行压栈的小于该值的元素都不会成为最小值,可以不压入辅助栈,这可以保证辅助栈栈顶即为数据栈中所有数据中的最小值。

函数说明: 压栈时,只有辅助栈为空,或该数值不大于辅助栈栈顶元素,才对辅助栈进行压栈。 出栈时,如果数据栈栈顶元素与辅助栈栈顶元素相等,说明该数值被弹出,也需要同时对辅助栈进行出栈。
核心代码

class Solution {
private:
    stack<int> sk;
    stack<int> help;
public:
    void push(int value) {
        sk.push(value);
        //保证辅助栈的单调性
        if(help.empty() || value <= help.top()) {
            help.push(value);
        }
    }
    void pop() {
        //如果辅助栈栈顶值与数据栈栈顶值相等则出栈
        if(help.top() == sk.top()) {
            help.pop();
        }
        sk.pop();
    }
    int top() {
        return sk.top();
    }
    int min() {
        return help.top();//辅助栈栈顶既为最小值
    }
};

复杂度分析
时间复杂度:,对单个操作,四个函数的时间复杂度均为常数级别
空间复杂度:,最坏情况下,辅助栈也需要存储所有元素,使用空间

方法二:单个栈
核心思想
也可以使用单个栈完成功能,但此时的栈不是存储数据,而是存储数据与当前最小值的差值,并另外存储一个最小值
函数说明: :压栈时,当栈为空时,存储最小值,压入差值0。栈不为空时,压入数据与当前最小值的差值,如果差值为负,则更新最小值 :出栈时,如果此时栈顶值为负数,说明此处发生的最小值的变化,需要更新最小值。 :如果栈顶值为负数,说明最小值在此处更新,此时返回最小值即可。如果栈顶值为正数,即为最小值与数据的差值,相加后进行返回 :直接返回存储的最小值即可
核心代码

class Solution {
private:
    stack<long> sk;//记录与最小值差值,由于可能大于INT_MAX,使用long
    long tmin;//记录最小值
public:
    void push(int value) {
        if(sk.empty()) {
            //当栈为空,最小值即为当前值,差值为0
            sk.push(0);
            tmin = value;
        } else {
            //压入与最小值的差值
            sk.push(value - tmin);
            if(value < tmin) {
                //如果当前值小于最小值,进行更新
                tmin = value;
            }
        }
    }

    void pop() {
        if(sk.top() < 0) {
            //差值为负数,说明当前位置最小值发生改变,需要改回该元素压入前的最小值
            tmin -= sk.top();
        }
        sk.pop();
    }

    int top() {
        if(sk.top() < 0) {
            return tmin;//此时最小值刚进行更新
        } else {
            return tmin + sk.top();
        }
    }
    int min() {
        return tmin;
    }
};

复杂度分析
时间复杂度:,对单个操作,四个函数的时间复杂度均为常数级别
空间复杂度:,只使用了常数个变量

全部评论

相关推荐

10-09 00:50
已编辑
长江大学 算法工程师
不期而遇的夏天:1.同学你面试评价不错,概率很大,请耐心等待;2.你的排名比较靠前,不要担心,耐心等待;3.问题不大,正在审批,不要着急签其他公司,等等我们!4.预计9月中下旬,安心过节;5.下周会有结果,请耐心等待下;6.可能国庆节前后,一有结果我马上通知你;7.预计10月中旬,再坚持一下;8.正在走流程,就这两天了;9.同学,结果我也不知道,你如果查到了也告诉我一声;10.同学你出线不明朗,建议签其他公司保底!11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务