题目:
设计一个支持 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.
思路:使用两个stack,一个data保存数据,一个min保存最小值,并且保持min的栈顶是最小值;
关键在于每次从栈顶压入数据时,data正常压入,而min则压入当前最小值。弹出数据时两个栈同时弹出;如图所示。
6 | -1 |
-1 | -1 |
0 | 0 |
3 | 1 |
2 | 1 |
1 | 1 |
data | min |
class MinStack {
public:
/** initialize your data structure here. */
MinStack(){}
~MinStack(){}
void push(int x) {
data.push(x);
if(min.empty()||x<min.top())
min.push(x);
else
min.push(min.top());
}
void pop() {
if(data.empty())
return;
min.pop();
data.pop();
}
int top() {
return data.top();
}
int getMin() {
return min.top();
}
private:
stack<int> data;
stack<int> min;
};