LintCode(M)带min函数的栈

带最小值操作的栈

描述
笔记
数据
评测
实现一个带有取最小值min方法的栈,min方法将返回当前栈中的最小值。

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

注意事项

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

您在真实的面试中是否遇到过这个题?
样例
如下操作:push(1),pop(),push(2),push(3),min(), push(1),min() 返回 1,2,1

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

    }

    void push(int number) {
        // write your code here
        m1.push(number);
        if(m2.empty()||number<=m2.top()) //最初m2.empty()写成m1.empty(),结果找了好久的bug...
        m2.push(number);
         else m2.push(m2.top()); 

    }

   int pop() {
        // write your code here
    if(m1.empty()) return -1;
        if(!m1.empty()){
          m2.pop();int m=m1.top();m1.pop(); 
           return m;

        }
        if (m1.empty()) return -1;
        int t = m1.top(); m1.pop();
        if (!m2.empty() && m2.top() == t) m2.pop();
        return t;
    }

    int min() {
        // write your code here
        if(!m2.empty()) 
        return m2.top();
        return -1;
    }
    private:
    stack<int> m1,m2;



};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章