定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

class Solution {
public:
    stack<int> s1,s2;
    void push(int value) {
        s1.push(value);
        if(s2.empty()){
            s2.push(value);
        }
        else{
            if(value<=s2.top()){
                s2.push(value);
            }
        }
    }
    void pop() {
        if(s1.top()==s2.top()){
            s2.pop();
        }
        s1.pop();
    }
    int top() {
        return s1.top();
    }
    int min() {
        return s2.top();
    }
};

思路:用一个辅助栈记录入栈元素从大到小,如果后面压入的元素比辅助栈顶元素大,就不用压入它(等于还是要,否则pop了会出错)。考虑635入栈,第一次s1:6,s2:6,min:6。第二次s1:6,3,s2:6,3,min:3,第三次s1:6,3,5,s2:6,3,min:3。pop时要注意如果pop的是最小值,s2也要pop