一、题目
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
- push(x) —— 将元素 x 推入栈中。
- pop() —— 删除栈顶的元素。
- top() —— 获取栈顶元素。
- getMin() —— 检索栈中的最小元素。
示例:
输入:
["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]
输出:
[null,null,null,null,-3,null,0,-2]
解释:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
提示:
pop、top 和 getMin 操作总是在 非空栈 上调用。
二、思路
vector用作栈容器,list维护最小元素值
首先可知在private中可使用一个数组vector<int> nums用于充当栈的容器,并用int size记录规模。由题目要求getMin()在常数时间内返回最小元素,因此可以在设计类时private设置一个成员变量min记录栈中的最小元素。对于push()方法,直接插入数组中,并分析只要值小于当前最小值min,则更新min。但是经分析pop()方法后发现,如果弹出的值恰好是最小值,那么min的值不能再是那个被弹出的值,应当恢复为曾经第二小的值。如果只使用一个int变量去维护min,那么它无法保存“第二小的值、第三小的值……”信息。因此可以使用一个链表(可以满足常数时间插入删除的要求)去维护“最小值”信息。一旦最小值被更新了,则将该值插入链表中即可。
class MinStack {
public:
MinStack() {
nums = vector<int>();
min = list<int>();
}
void push(int val) {
nums.push_back(val);
if (min.empty())//如果min尚且为空,
min.push_front(val);//直接存入
else if (val <= min.front())//只要val小于当前最小值(min中的首元素),
min.push_front(val);//将其插入当前首元素之前作为新的首元素
}
void pop() {
if (nums.empty())
return;//如果栈为空,不予任何操作
if (nums.back() == min.front())//如果栈顶的值恰为当前最小值,
min.erase(min.begin());//则删除链表min的首节点(当前将要被弹出的最小值)以让原来第二小的元素作为min首节点
nums.pop_back();//从栈中弹出栈顶
}
int top() {
return nums.back();
}
int getMin() {
return min.front();//链表首元素即为当前最小值
}
private:
vector<int> nums;
list<int> min;//使用链表存储维护“最小值”信息
};
/**
* Your MinStack object will be instantiated and called as such:
* MinStack* obj = new MinStack();
* obj->push(val);
* obj->pop();
* int param_3 = obj->top();
* int param_4 = obj->getMin();
*/
三、官方题解(来源:力扣(LeetCode))
辅助栈
要做出这道题目,首先要理解栈结构先进后出的性质。
对于栈来说,如果一个元素 a 在入栈时,栈里有其它的元素 b, c, d,那么无论这个栈在之后经历了什么操作,只要 a 在栈中,b, c, d 就一定在栈中,因为在 a 被弹出之前,b, c, d 不会被弹出。
因此,在操作过程中的任意一个时刻,只要栈顶的元素是 a,那么我们就可以确定栈里面现在的元素一定是 a, b, c, d。
那么,我们可以在每个元素 a 入栈时把当前栈的最小值 m 存储起来。在这之后无论何时,如果栈顶元素是 a,我们就可以直接返回存储的最小值 m。
算法
按照上面的思路,我们只需要设计一个数据结构,使得每个元素 a 与其相应的最小值 m 时刻保持一一对应。因此我们可以使用一个辅助栈,与元素栈同步插入与删除,用于存储与每个元素对应的最小值。
- 当一个元素要入栈时,我们取当前辅助栈的栈顶存储的最小值,与当前元素比较得出最小值,将这个最小值插入辅助栈中;
- 当一个元素要出栈时,我们把辅助栈的栈顶元素也一并弹出;
- 在任意一个时刻,栈内元素的最小值就存储在辅助栈的栈顶元素中。
class MinStack {
stack<int> x_stack;
stack<int> min_stack;
public:
MinStack() {
min_stack.push(INT_MAX);
}
void push(int x) {
x_stack.push(x);
min_stack.push(min(min_stack.top(), x));
}
void pop() {
x_stack.pop();
min_stack.pop();
}
int top() {
return x_stack.top();
}
int getMin() {
return min_stack.top();
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/min-stack/solution/zui-xiao-zhan-by-leetcode-solution/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
复杂度分析
- 时间复杂度:对于题目中的所有操作,时间复杂度均为 O(1)。因为栈的插入、删除与读取操作都是 O(1),我们定义的每个操作最多调用栈操作两次。
- 空间复杂度:O(n),其中 n 为总操作数。最坏情况下,我们会连续插入 n 个元素,此时两个栈占用的空间为 O(n)。