问题描述
设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
- push(x) – 将元素 x 推入栈中。
- pop() – 删除栈顶的元素。
- top() – 获取栈顶元素。
- getMin() – 检索栈中的最小元素。
解题报告
这个最小栈和一般的栈的区别在于其能够直接返回最小值。
我们可以使用一个变量来保存最小值,但如果仅仅只是这么做,那么当这个最小值弹出时,第二小的值就无法找到了。
所以,在每次更新最小值时【必须是x<=MINN
,这里需要取等号,如果这里不取等号,在其他位置的最小值出栈时,函数会将其他的非最小值更新为最小值】,我们首先将旧的最小值压入栈中,然后压入最小值;而当弹出的元素为最小值时,我们同时将栈顶元素的下一个元素也弹出,将第二次弹出的元素赋值为当前的最小值。
实现代码
class MinStack {
public:
/** initialize your data structure here. */
stack<int>stack;
int MINN=INT_MAX;
MinStack() {
}
void push(int x) {
if(x<=MINN){
stack.push(MINN);
MINN=x;
}
stack.push(x);
}
void pop() {
if(stack.top()==MINN){
stack.pop();
MINN=stack.top();
}
stack.pop();
}
int top() {
return stack.size()==0?-1:stack.top();
}
int getMin() {
return MINN;
}
};
/**
* 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();
*/
参考资料
[1] Leetcode 155.最小栈
[2] 题解区:windliang