leetcode155. 最小栈

leetcode155. 最小栈

设计一个支持 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.

提示：

pop、top 和 getMin 操作总是在 非空栈 上调用。

解题思路

• 常规写法（超时）

class MinStack {
public:
    stack<int> st;
    MinStack() {
    }
    
    void push(int x) {
        st.push(x);
    }
    
    void pop() {
        st.pop();
    }
    
    int top() {
        return st.top();
    }
    
    int getMin() {
        //辅助栈
        stack<int> tmp;
        int minn = st.top();
        //取最小值
        while(!st.empty()){
            if(minn>st.top())
                minn = st.top();
            tmp.push(st.top());
            st.pop();
        }
        
        while(!tmp.empty()){
            st.push(tmp.top());
            tmp.pop();
        }
        
        return minn;
    }
};

• 构建辅助栈，与主栈同时修改

class MinStack {
public:
    stack<int> st;
    stack<int> Min;//栈顶元素始终存储st中最小值
    const int max_int = 0x7fffffff;//设置为int型最大值2147483647
    MinStack() {
        Min.push(max_int);//初始化
    }
    
    void push(int x) {
        st.push(x);
        if(x<=Min.top())
            Min.push(x);
    }
    
    void pop() {
        if(st.top()==Min.top())
            Min.pop();
        st.pop();
    }
    
    int top() {
        return st.top();
    }
    
    int getMin() {
        return Min.top();
    }
};

执行用时：44 ms, 在所有 C++ 提交中击败了76.64% 的用户
内存消耗：14.6 MB, 在所有 C++ 提交中击败了45.50% 的用户