设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/min-stack
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
人尽皆知的一道题目,不过一般都是两个栈的做法,这里记录一个O(1)空间的做法。
维护一个栈,和一个最小值。这里栈存放的值是每次data与当前Minvalue的差值。这样每次就可以记录这两个值了。每次diff = data-Minvalue,如果diff < 0就更新MinValue = data。对于pop的操作,判断每次top是否大于0,如果大于,就表明当前的Minvalue是之前的值,不用更新,否则当前的MinValue就是这一次插入的data,则上一次的MinValue = Minvalue-st.top()
这里用longlong,是因为LC的这道题目,会爆int。
对于最大栈的问题,这里思路是相同的。
具体代码如下:
class MinStack {
public:
typedef long long ll;
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
if(st.empty()){
MinValue = x;
st.push(0);
}
else{
ll diff = x - MinValue;
st.push(diff);
MinValue = (diff>0? MinValue:x);
}
}
void pop() {
if(st.empty()){
return;
}
MinValue = (st.top()>0? MinValue:MinValue-st.top());
st.pop();
}
int top() {
return (st.top()>0? MinValue+st.top():MinValue);
}
int getMin() {
return MinValue;
}
private:
stack<ll> st;
ll MinValue;
};
/**
* 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();
*/