剑指 Offer 30. 包含min函数的栈

剑指 Offer 30. 包含min函数的栈

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

解题思路

使用了一个的栈和一个变量。
思路大概是，栈内存放元素与最小值的差，用一个变量存放最小值
入栈的时候，如果差是大于等于零的，说明该元素不会更新最小值，直接将差值入栈，最小值不变；如果差是小于零的，说明该元素会更新最小值，将该差值入栈，然后将最小值更新为该元素。
出栈的时候，如果栈顶元素大于等于零，说明栈顶元素对应的不是最小值，直接pop即可；如果栈顶元素小于零，说明栈顶元素对应的是最小值，需要先把最小值更新为当前最小值减去栈顶元素，然后再pop。
返回top时，先判断栈顶元素，如果小于零，直接返回最小值；如果大于等于零，则返回最小值加栈顶元素。
返回最小值时，直接返回即可。
栈和变量的类型为long，因为用int的时候发现会溢出，原因是入栈的元素实际上是差值，当最小值为int的最小值时，入栈的差值可能超出int的范围，所以用了long,但不知这样是否可以。

代码

class MinStack {
public:
    /** initialize your data structure here. */
    stack<long> s;
    long minNum;

    MinStack() {
    }
    
    void push(long x) {
        if(s.empty()==true){
            minNum=x;
            s.push(0);
        }
        else {           
            s.push(x-minNum);
            minNum=x-minNum>=0?minNum:x;            
        }
        
    }
    
    void pop() {
        minNum=s.top()>=0?minNum:minNum-s.top();
        s.pop();
    }
    
    int top() {
        if(s.top()<0){
            return minNum;
        }
        else{
            return s.top()+minNum;
        }
        
    }
    
    int min() {
        return minNum;
    }
};

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