题目链接
Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.
push(x) – Push element x onto stack.
pop() – Removes the element on top of the stack.
top() – Get the top element.
getMin() – Retrieve the minimum element in the stack.
Example:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> Returns -3.
minStack.pop();
minStack.top(); --> Returns 0.
minStack.getMin(); --> Returns -2.
题意:
设计一个最小栈,实现push, pop, top, getMin四个功能。这道题和普通的栈多了一个功能,可以返回当前栈内的最小值。
题解:
使用两个栈,栈1记录进来的数,栈2记录目前的最小值。当有新数push进来的时候,如果栈2为空或者这个数小于栈2顶上的值,就把这个数推入栈2。当pop的数正好等于最小值时,说明当前栈内的最小值变化了,要弹出这个最小值,记录的下一个最小值来到栈顶。
class MinStack {
public:
/** initialize your data structure here. */
stack <int> stackValue;
stack <int> stackMin;
MinStack() {
}
void push(int x) {
stackValue.push(x);
if(stackMin.empty() || x <= stackMin.top())
stackMin.push(x);
}
void pop() {
int x = stackValue.top();
stackValue.pop();
if(stackMin.top() == x)
stackMin.pop();
}
int top() {
return stackValue.top();
}
int getMin() {
return stackMin.top();
}
};