LintCode(十二):带最小值操作的栈

描述

实现一个带有取最小值min方法的栈,min方法将返回当前栈中的最小值。

你实现的栈将支持push,pop 和 min 操作,所有操作要求都在O(1)时间内完成。

如果堆栈中没有数字则不能进行min方法的调用

您在真实的面试中是否遇到过这个题?

样例
如下操作:push(1),pop(),push(2),push(3),min(), push(1),min() 返回 1,2,1

思路:用两个栈分别来存放当前输入元素【栈data】,以及当前栈里最小的元素【栈mins】(注意每一次都要往mins里入一个数)

C++实现:

class MinStack {
public:

    std::stack<int> date;
    stack<int> mins;
    MinStack() {
        // do intialization if necessary
    }

    /*
     * @param number: An integer
     * @return: nothing
     */
    void push(int number) {
        // write your code here
        date.push(number);
        if(mins.size()==0 || mins.top()>=number)
        {
            mins.push(number);
        }
        
        else{
            mins.push(mins.top());
        }
        
    }

    /*
     * @return: An integer
     */
    int pop() {
        // write your code here
        if(mins.size()>0 && date.size()>0)
        {
            int t = date.top();
            date.pop();
            mins.pop();
            return t;
        }
        else{
            return 0;
        }
    }

    /*
     * @return: An integer
     */
    int min() {
        // write your code here
        if(mins.size()>0 && date.size()>0)
        {
            return mins.top();
        }
        
        return 0;
    }
};

Py3:

class MinStack(object):

    def __init__(self):
        # do some intialize if necessary
        self.stack = []

    def push(self, number):
        # write yout code here
        if len(self.stack) == 0:
            self.stack.append([number, number])
        else:
            self.stack.append([number, min(number, self.stack[-1][1])])

    def pop(self):
        # pop and return the top item in stack
        return self.stack.pop()[0]

    def min(self):
        # return the minimum number in stack
        return self.stack[-1][1]

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值