Leetcode 155. 最小栈 O(1)空间复杂度的做法

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

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

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/min-stack
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

人尽皆知的一道题目,不过一般都是两个栈的做法,这里记录一个O(1)空间的做法。

维护一个栈,和一个最小值。这里栈存放的值是每次data与当前Minvalue的差值。这样每次就可以记录这两个值了。每次diff = data-Minvalue,如果diff < 0就更新MinValue = data。对于pop的操作,判断每次top是否大于0,如果大于,就表明当前的Minvalue是之前的值,不用更新,否则当前的MinValue就是这一次插入的data,则上一次的MinValue = Minvalue-st.top()

这里用longlong,是因为LC的这道题目,会爆int。
对于最大栈的问题,这里思路是相同的。
具体代码如下:


class MinStack {
public:
    typedef long long ll;
    /** initialize your data structure here. */
    MinStack() {
        
    }
    
    void push(int x) {
        if(st.empty()){
            MinValue = x;
            st.push(0);
        }
        else{
            ll diff = x - MinValue;
            st.push(diff);
            MinValue = (diff>0? MinValue:x);
        }
    }
    
    void pop() {
        if(st.empty()){
            return;
        }
        MinValue = (st.top()>0? MinValue:MinValue-st.top());
        st.pop();
    }
    
    int top() {
        return (st.top()>0? MinValue+st.top():MinValue);
    }
    
    int getMin() {
        return MinValue;
    }
private:
    stack<ll> st;
    ll MinValue;
};

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

你可能感兴趣的:(leetcode,leetcode)