实现一个栈,提供三种操作:min、pop、push,其中min方法返回栈中最小数,要求空间占用与时间占用最低。
分析
由栈的性质来看,如果一个数的后面没有任何数,这个数才能被删除。
那么操作:
- push:如果这个数比最小数小,那么这个数成为最小数;其余情况(大于或等于最小数)则最小数不变。
- pop:如果这个数是(注意“是”不是“等于”)最小数,那么最小数回滚到上一个最小数。
这么看来,最小数也可以作为一个栈来维护,注意维护的不是最小数的值,而是最小数的位置(当然这样会破坏栈调用的原则-.-)。
实现
使用两个栈,第一个栈记录所有数,第二个栈记录最小数的位置,
代码
class MinStack {
public:
vector<int> mt;
vector<int> st;
MinStack() { }
void push(int x) {
if (mt.empty() || st.at(mt.back()) > x) {
mt.push_back(st.size());
}
st.push_back(x);
}
void pop() {
st.pop_back();
if (mt.back() == st.size()) {
mt.pop_back();
}
}
int top() {
return st.back();
}
int getMin() {
return st.at(mt.back());
}
};
补充
在leetCode上面看到不少优秀的实现方式,其中用的最多的是双栈实现(入栈时同时保存一个最小数的栈),这种方式比较简单也满足一般用途。
如果需要优化,可以考虑使用自定义的栈来替代vector,具体实现可以根据需要。