一个栈是没有办法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