使用一个保存数据的栈和一个保存当前最小值的辅助栈,辅助的栈顶元素总是目前的最小值,在push和pop的时候维护两个栈就可以了。
class Solution {
public:
void push(int value) {
stk.push(value);
if(min_stk.size()==0 || min_stk.top()>value) min_stk.push(value);
else min_stk.push(min_stk.top());
}
void pop() {
stk.pop();
min_stk.pop();
}
int top() {
return stk.top();
}
int min() {
return min_stk.top();
}
private:
stack<int> stk{};
stack<int> min_stk{};
};