面试题30:包含min函数的栈

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

class Solution {
public:
    //与书中所写思路一致:最小栈栈顶即为当前最小值
    void push(int value) 
    {   //若栈为空,则压入栈st,min_st
        if(st.empty())
        {
            st.push(value);
            min_st.push(value);
        }
        else
        {
            st.push(value);
            //若新入栈的元素小于最小栈栈顶元素,则此元素为新的最小元素,入最小栈
            if(st.top()<min_st.top())
                min_st.push(value);
            //若新入栈的元素大于最小栈栈顶元素,则最小元素不变
            else
                min_st.push(min_st.top());
        }
    }
    //出栈处理一样:栈顶与最小栈同时出栈即可
    void pop() 
    {
        if(st.empty())
            return;
        st.pop();
        min_st.pop();

    }
    //直接取st栈顶元素
    int top() {
        if(st.empty())
            return NULL;
        return st.top();
    }

    int min() {
        if(min_st.empty())
            return NULL;
        return min_st.top();
    }
private:
    //创建两个栈,st正常存放入栈元素,min_st存放最小元素
    stack<int> st;
    stack<int> min_st;
};
全部评论

相关推荐

不要停下啊:大二打开牛客,你有机会开卷了,卷起来,去找课程学习,在牛客上看看大家面试笔试都需要会什么,岗位有什么需求就去学什么,努力的人就一定会有收获,这句话从来都经得起考验,像我现在大三了啥也不会,被迫强行考研,炼狱难度开局,啥也不会,找工作没希望了,考研有丝丝机会
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
昨天 15:37
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务