访问栈中最小数

实现一个栈,提供三种操作: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,具体实现可以根据需要。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值