题目
设计一个支持 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.
思路
- 使用两个栈,一个栈用来正常的push ,pop ,top 操作。
- 另一个栈,当栈为空时,元素入栈。当当前元素大于栈顶元素时,不入栈。当前元素小于栈顶元素时,入栈,表示当前最小元素。当pop操作的时候,如果第一个栈的栈顶元素等于第二个栈的栈顶元素,则第二个栈的栈顶元素也出栈。
代码
class MinStack {
private:
stack<int> mainStack;
stack<int> minStack;
public:
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
mainStack.push( x );
if ( minStack.empty() )
minStack.push( x );
else {
if ( x <= minStack.top() )
minStack.push( x );
}
return;
}
void pop() {
if ( mainStack.top() == minStack.top() )
minStack.pop();
mainStack.pop();
return;
}
int top() {
return mainStack.top();
}
int getMin() {
return minStack.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();
*/