每日一题——实现取最小值的栈

实现一个栈,要求实现push,pop,min(返回最小值)的时间复杂度均为O(1)

思路:使用两个栈,一个是datastack,用来存放每次push进的数据,而另一个是minstack,专门存放当前的最小值。
压栈:如果datastack为空或插入的值小于minstack中的栈顶元素,说明该值是最小值,所以向两个栈中都插入该值;至于其他情况,说明该值不是最小值,所以将该值插入datastack,而将minstack的栈顶元素再插一遍进minstack,保证minstack的栈顶元素始终是最小值。

出栈:两个栈同时出栈,保证两个栈的元素个数始终保持相等;返回datastack的栈顶元素;
取最小值;直接返回minstack的栈顶元素即可。

class MinStack {
public:
    stack<int> minstack;
    stack<int> datastack;
    int ret = 0;
    MinStack() {
        // do initialization if necessary
    }

    void push(int number) {
        // write your code here
        if(datastack.empty() || number < minstack.top())
        {
            datastack.push(number);
            minstack.push(number);
        }
        else
        {
            datastack.push(number);
            minstack.push(minstack.top());
        }
    }

    int pop() {
        // write your code here
        ret = datastack.top();
        datastack.pop();
        minstack.pop();
        return ret;
    }

    int min() {
        // write your code here
        return minstack.top();
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值