除了维护基本的栈s以外,还有额外维护一个存放当前栈中最小元素的栈ms。假设当前需要入栈s的元素为x,则当ms为空或者x<=ms.top()时,x也要压入到ms中。假设当前栈s顶部元素为x,则出栈时若x==ms.top()则ms也要进行出栈操作。
需要注意的是,当最小栈的入栈条件中x必须要小于等于ms.top(),不可以是严格小于。否则的话当多个最小元素入栈后,再进行出栈操作会出错。
具体代码如下:
class MinStack {
public:
/** initialize your data structure here. */
stack<int> s; //存放元素
stack<int> ms; //存放当前栈中最小元素
MinStack() {
}
void push(int x) {
if(ms.empty() || x <= ms.top()) ms.push(x);
s.push(x);
}
void pop() {
if(s.top() == ms.top()) ms.pop();
s.pop();
}
int top() {
return s.top();
}
int getMin() {
return ms.top();
}
};
/**
* 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.getMin();
*/