例题描述
设计一个支持 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();
}
};