设计一个支持 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.
解题思路
通过双栈来实现最小栈,因为这样可以在常数时间复杂度中得到栈内元素最小值。这是典型的空间换时间的操作。
假定栈内元素按顺序分别为:3、2、5、1、7。
压栈
入栈时要对两个栈同时进行操作,normal栈正常压栈,对于min栈进行参数条件判定:如果此时
最小栈为空 (说明是第一次压栈)
这个值小于此时最小栈的栈顶元素 (说明是压入了一个小于已入栈全部元素的值)
两条件是任满足其一即可进行最小栈的压栈,否则不满足条件执行最小栈的栈顶元素压栈,说明压入的这个值不是最小值,将目前最小值再次压入最小栈。
例如5元素压栈时,此时5不小于已入栈的任意一个元素,他不是最小值,所以将最小栈min的栈顶元素进行二次压栈。
最后入栈情况如图所示,最小栈中存放的是每次压栈时的最小值:
弹栈
弹出normal栈栈顶元素的同时也要弹出min栈的栈顶元素。因为如果移除了普通栈的栈顶元素,此时栈内最小值可能发生了变化,比如弹出1时,最小值就变成了2,所以也要将最小栈中栈顶元素1弹出。以此类推,此时最小值仍为1。
获取栈顶元素
此操作与最小值无关,也不会影响到栈内元素的顺序,所以选择直接返回normal栈的栈顶元素值即可。
检索栈中最小值
这时我们的最小栈就可以崭露头角了。
直接返回最小栈min的栈顶元素即可。因为无论何时min栈的栈顶元素都存放这此时normal栈内最小元素值。
代码实现
class MinStack {
public:
stack<int> normal;
stack<int> min;
//1. 入栈
void push(int x) {
normal.push(x);
if(min.empty() || x <= min.top()){
min.push(x);
}
else{
min.push(min.top());
}
}
//2. 出栈
void pop() {
normal.pop();
min.pop();
}
//3. 获取栈顶元素
int top() {
return normal.top();
}
//4. 检索栈中的最小元素
int getMin() {
return min.top();
}
};