面试题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;
};
全部评论

相关推荐

点赞 评论 收藏
分享
牛客5655:其他公司的面试(事)吗
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务