包含min函数的栈

题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

自己的思路
由于考虑到时间复杂度要为O(1),所以就需要能直接取出最小值,试想,如果栈顶元素就是最小值时,就可以解决这个问题了,于是考虑使用两个栈,一个是最开始的入栈Stack1,不做处理,另一个栈Stack2每次通过比较,将第一个栈的最小值不断地压栈,最后将栈顶出栈就好了,但是可能是有条件考虑出错,导致不能a,先把错误代码贴出,后面再修改

class Solution {
public:
    void push(int value) {
        StackInt.push(value);
        if(StackMin.empty())
            StackMin.push(value);
        else if(value<StackMin.top())
            StackMin.push(value);
    }
    void pop() {
        StackInt.pop();
    }
    int top() {
        return StackInt.top();
    }
    int min() {
        return StackMin.top();
    }
private:
    stack<int> StackInt;
    stack<int> StackMin; //通过将StackMin的栈顶置为StackInt中的最小值,就可以是O(1)取出栈的最小值
};

大佬的思路
由于没有考虑出栈的情况,导致出错,所以还是要把出栈也考虑进去,不可能一直做压栈处理,那也太简单了,要保证每次出栈之后的栈顶元素始终是当前栈中的最小元素(两个栈的元素个数要相等)。(这也是解这道题的关键和细节所在)
通过一个策略,如果压栈的元素比StackMin的top元素大,则将StackMin的top元素再压栈一次,最终得到下面的两个栈,而每次出栈,只需要弹出栈顶元素就可以了,且StackMin的top永远是最小的,很妙。
在这里插入图片描述

class Solution {
public:
    void push(int value) {
        StackInt.push(value);
        if(StackMin.empty())
            StackMin.push(value);
        else if(StackMin.top()<value)
            StackMin.push(StackMin.top()); //为了保证两个栈的数据个数相等(这也是一个细节)
        else
            StackMin.push(value);
    }
    void pop() {
        if(!StackInt.empty()){ //出栈的时候每次都将两个栈的栈顶出栈顶出栈就好了
            StackInt.pop();
            StackMin.pop();
        }
    }
    int top() {
        return StackInt.top();
    }
    int min() {
        return StackMin.top();
    }
private:
    stack<int> StackInt;
    stack<int> StackMin; //通过将StackMin的栈顶置为StackInt中的最小值,就可以是O(1)取出栈的最小值
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值