1. 题目来源
链接:包含min函数的栈
来源:LeetCode——《剑指-Offer》专项
2. 题目说明
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min
函数在该栈中,调用 min
、push
及 pop
的时间复杂度都是
O
(
1
)
O(1)
O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
提示:
- 各函数的调用总次数不超过 20000 次
3. 题目解析
方法一:模拟+辅助栈,常规解法
这个最小栈跟原来的栈相比就是多了一个功能:可以返回该栈的最小值。采用模拟做思路还是很明确的:使用两个栈来实现,一个栈来按顺序存储 push 进来的数据,另一个用来存出现过的最小值。
这个代码没经过啥步骤优化,写的比较直白,中间的 if - else
其实能够进行优化,仅做参考。
参见代码如下:
// 执行用时 :28 ms, 在所有 C++ 提交中击败了75.09%的用户
// 内存消耗 :17.7 MB, 在所有 C++ 提交中击败了100.00%的用户
class MinStack {
public:
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
if (s1.empty()) {
s1.push(x);
s2.push(x);
}
else
s1.push(x);
if (s1.top() >= s2.top())
s2.push(s2.top());
else
s2.push(s1.top());
}
void pop() {
s1.pop();
s2.pop();
}
int top() {
return s1.top();
}
int min() {
return s2.top();
}
private:
stack<int> s1;
stack<int> s2;
};
/**
* Your MinStack object will be instantiated and called as such:
* MinStack* obj = new MinStack();
* obj->push(x);
* obj->pop();
* int param_3 = obj->top();
* int param_4 = obj->min();
*/
方法二:一个栈+整形变量,巧妙解法
这种解法比较巧妙,只用到了一个栈,还需要一个整型变量 min_val
来记录当前最小值,初始化为整型最大值,如果需要进栈的数字小于等于当前最小值 min_val
,则将 min_val
压入栈,并且将 min_val
更新为当前数字。在出栈操作时,先将栈顶元素移出栈,再判断该元素是否和 min_val
相等,相等的话将 min_val
更新为新栈顶元素,再将新栈顶元素移出栈即可。
简单概括:
- 当有更小的值的时候,先把以前的最小值压入,然后再压入新的最小值
- 弹出的时候,假如弹出的值和当前的最小值是一样的,说明下一个值是之前的最小值,再弹出一次取到这个值
这个说起来比较抽象,建议画个图,走一遍测试用例就能体会到它的巧妙了。
参见代码如下:
class MinStack {
public:
/** initialize your data structure here. */
MinStack() {
min_val = INT_MAX;
}
void push(int x) {
if (x <= min_val) {
st.push(min_val);
min_val = x;
}
st.push(x);
}
void pop() {
int t = st.top(); st.pop();
if (t == min_val) {
min_val = st.top(); st.pop();
}
}
int top() {
return st.top();
}
int min() {
return min_val;
}
private:
int min_val;
stack<int> st;
};
/**
* Your MinStack object will be instantiated and called as such:
* MinStack* obj = new MinStack();
* obj->push(x);
* obj->pop();
* int param_3 = obj->top();
* int param_4 = obj->min();
*/