【LeetCode】155. 最小栈

155. 最小栈（中等）

方法一：辅助栈

思路

• 题目要求在常数时间内获得栈中的最小值，因此不能在的时候再去计算getMin0最小值，最好应该在 push 或者 pop 的时候就已经计算好了当前栈中的最小值。
• 我在 push 和 pop 操作都设置了「辅助栈」。St 用来保存正常输入顺序的数字， minSt 在 栈 temp 的帮助下，始终按照从小到大的顺序（即 minSt.top()) 是最小元素）排列。
• 这种方法下， St 和 minSt 的数据始终是一样的。因此并不是一个好的思路，请看方法二。

代码

class MinStack {
public:
    stack<int> St, minSt;
    MinStack() {

    }
    
    void push(int val) {
        St.push(val);
        stack<int> temp;
        if(minSt.empty()){
            minSt.push(val);
        }
        else{
            while(!minSt.empty()){
                if(minSt.top() >= val){
                    minSt.push(val);
                    break;
                }
                else{
                    temp.push(minSt.top());
                    minSt.pop();
                }  
            }
            while(!temp.empty()){
                minSt.push(temp.top());
                temp.pop();
            }
        }
    }
    
    void pop() {
        int head = St.top();
        St.pop();
        stack<int> temp;
        while(!minSt.empty()){
            if(minSt.top() != head){
                temp.push(minSt.top());
                minSt.pop();
            }
            else{
                minSt.pop();
                break;
            }
        }               

        while(!temp.empty()){
            minSt.push(temp.top());
            temp.pop();
        }
    }
    
    int top() {
        return St.top();
    }
    
    int getMin() {
        return minSt.top();
    }
};

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

方法二：辅助栈优化

思路

• 方法一用到了三个栈，其中 St 和 minSt 的数据始终是一样的，当数据栈的数越来越多的时候，我们要在辅助栈minSt 的栈顶放置和 栈St栈顶一样的元素，这样做有点“浪费”。基于这一点，我们做一些“优化”。

  

  

代码

class MinStack {
public:
    stack<int> St, minSt;
    MinStack() {
        minSt.push(INT_MAX);
    }
    
    void push(int val) {
        St.push(val);
        if(val <= minSt.top()){
            minSt.push(val);
        }else{
            minSt.push(minSt.top());
        }
    }
    
    void pop() {
        St.pop();
        minSt.pop();
    }
    
    int top() {
        return St.top();
    }
    
    int getMin() {
        return minSt.top();
    }
};

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