LeetCode 155. Min Stack【栈+设计/辅助栈+空间换时间】⭐⭐⭐

题目描述

知识点

栈、设计

结果

实现(写错了。。。)

## 码前思考

1. 使用vector来实现栈；
2. 使用priority_queue来实时的保存最小值；
   ## 代码实现

class MinStack {
public:
    vector<int> st;
    priority_queue<int,vector<int>,greater<int>> q; //使用优先队列存储最小值
    int minVal;    //存储最小值
    /** initialize your data structure here. */
    MinStack() {
        minVal = INT_MAX;  //初始化为最大值
    }
    
    void push(int x) {
        minVal = minVal > x ? x : minVal;
        q.push(x);
        st.push_back(x);
    }
    
    void pop() {
        int backVal = st.back();
        //如果弹出的是最小值
        if(minVal == backVal){
            q.pop();
            minVal = q.top();
        }
        st.pop_back();
    }
    
    int top() {
        return st.back();//返回top的值
    }
    
    int getMin() {
        return minVal;
    }
};
/**
 * 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();
 */

## 码后反思

1. 题目中有要求，查询最小值要在$O(1)$时间内完成~所以要pop(),push()的时候实时调整最小值！
2. 获取vector的最后一个元素使用back()；
3. 将priority_queue改成最小堆：priority_queue<int,vector<int>,greater<int>>；~~
4. 看了官方题解之后，发现最nice的方法是使用 辅助栈 的思想(这是一种典型的用空间换时间的方法~~)，运行结果和代码如下：
   

class MinStack {
public:
    stack<int> st;
    stack<int> minSt;

    /** initialize your data structure here. */
    MinStack() {
        minSt.push(INT_MAX); //初始化为最大值
    }
    
    void push(int x) {
        st.push(x);
        if(x > minSt.top()){
            minSt.push(minSt.top());
        }else{
            minSt.push(x);
        }
    }
    
    void pop() {
        st.pop();
        minSt.pop();
    }
    
    int top() {
        return st.top();//返回top的值
    }
    
    int getMin() {
        return minSt.top();
    }
};

/**
 * 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();
 */