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;



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