一个栈是没有办法O(1)弹出最小值的,所以需要设置一个辅助栈来随时取出目前的最小值。
GetMin(): 辅助栈的栈顶元素表示当前的最小值
push(int x): 当插入一个元素的时候,和辅助栈的栈顶元素比较,如果插入的元素<=当前栈顶元素的话,就把这个元素也插入到辅助栈中
pop(): 当弹出一个元素的时候,如果弹出的元素==栈顶元素的话,就把辅助栈的栈顶元素也弹出
c++

class Solution {
public:
    vector<int> getMinStack(vector<vector<int> >& op) {
        stack<int> S;
        stack<int> minS;
        vector<int> ans;
        for(int i = 0 ; i < op.size() ; i++)
        {
            if(op[i][0]==1)//插入
            {
                S.push(op[i][1]);
                if(minS.empty()||op[i][1] <= minS.top())
                {
                    minS.push(op[i][1]);
                }
            }
            else if(op[i][0]==2)//弹出
            {
                int t= S.top();
                S.pop();
                if(t == minS.top())
                {
                    minS.pop();
                }
            }
            else if(op[i][0]==3)//Getmin
            {
                ans.push_back(minS.top());
            }
        }
        return ans;
    }
};

java


python