155. 最小栈

题目描述

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

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

 

思路

利用一个辅助栈来记录最小元素，辅助栈的栈顶元素为当前栈的最小元素。也就是当push一个数时，若此数比辅助栈的栈顶元素要小，就把这个数压入辅助栈；否则压入辅助栈栈顶元素。

 

代码

class MinStack {
public:
    /** initialize your data structure here. */
    MinStack() {
    }
    
    void push(int x) {
        if (help.empty()){
            help.push(x);
        }
        else{
            if (x <= help.top()){
                help.push(x);
            }
            else{
                help.push(help.top());
            }
        }
        myStack.push(x);
    }
    
    void pop() {
        myStack.pop();
        help.pop();
    }
    
    int top() {
        return myStack.top();
    }
    
    int getMin() {
        return help.top();
    }
    
private:
    stack<int> help;
    stack<int> myStack;
};

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack* obj = new MinStack();
 * obj->push(x);
 * obj->pop();
 * int param_3 = obj->top();
 * int param_4 = obj->getMin();
 */