lintcode 第12题

/*

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

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

*/

#include<assert.h>
class MinStack {  


public:
    MinStack() {
        // do initialization if necessary
    }


    void push(int number) {
        // write your code here
        if(mindata.size()==0||mindata.top()>number)
        mindata.push(number);
        else
        mindata.push(mindata.top());
        data.push(number);
    }


    int pop() {
        // write your code here
        assert(mindata.size()>0&&data.size()>0);
        int temp=data.top();
        data.pop();
        mindata.pop();
        return temp;
    }


    int min() {
        // write your code here
        assert(mindata.size()>0&&data.size()>0);// assert的作用是现计算表达式 expression ，如果其值为假（即为0），那么它先向stderr打印一条出错信息，然后通过调用 abort 来终止程序运行。
        return mindata.top();
    }
private:
    stack<int> mindata,data;//mindata是一个辅助栈
};

如有问题请留言。

如有帮助请顶一个，你们的支持是我最大的动力。

文章均可以转载，但请注明文章链接，谢谢。