Leetcode155. 最小栈(HOT100)

kitesxian2024-11-25 20:40

代码：

cpp 复制代码

class MinStack {
private:
    stack<int> f,s;
public:
    MinStack(){}
    
    void push(int val) {
        s.push(val);
        if(f.empty()||val<=f.top()){
            f.push(val);
        }
    }
    
    void pop() {
        if(s.top()<=f.top())f.pop();
        s.pop();
    }
    
    int top() {
        return s.top();
    }
    
    int getMin() {
        return f.top();
    }
};

题解：这个题意还是比较难懂，代码写起来很简单，我们定义类的私有变量s作为存储数据的栈，然后为了常数时间内返回栈中最小元素，所以我们还定义了一个栈 f。f[i] 即：与之对应的s栈中目前最小的元素，但是我们还做了一个优化：如果你push进s栈一个很大的元素，我们 f 栈没必要动弹，因为我们f 记录的是最小元素。怎么判断？你push进s栈的元素如果小于等于f 的栈顶，我才在f 栈加一个元素。否则我f 栈什么都不做。

pop的时候也一样，pop的元素如果大于我的栈顶元素，那f 栈就不必更新什么。反之，如果你pop的元素小于等于f 栈顶元素，那么说明我之前记录过，所以f 也要pop。