1. 程式人生 > >lintcode12. 帶最小值操作的棧

lintcode12. 帶最小值操作的棧

最近研究逆向,再加上考試,一直沒刷題。逆向算是基本入門了,過幾天有空寫一篇入門級的逆向分享出來。

實現一個帶有取最小值min方法的棧,min方法將返回當前棧中的最小值。

你實現的棧將支援pushpop  min 操作,所有操作要求都在O(1)時間內完成。

 注意事項

如果堆疊中沒有數字則不能進行min方法的呼叫

樣例

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

思路:  要求在o(1)時間內完成,證明求min時不應該與棧的大小有關。

設計一個輔助棧,每push一個數時,同時將當前最小值壓入棧中,即記錄當前棧狀態時的最小值,有點動態規劃的意思。


程式碼:

class MinStack {
public:
    stack<int> a;
    stack<int> minstack;
    int minnumber;
    MinStack() {
        minnumber=0x7fffffff;
      // stack<int> a; // do intialization if necessary
    }

    /*
     * @param number: An integer
     * @return: nothing
     */
    void push(int number) {
        a.push(number);
        if(number<minnumber) minnumber=number;
        minstack.push(minnumber);
        // write your code here
    }

    /*
     * @return: An integer
     */
    int pop() {
         int c=a.top();
         a.pop();
         minstack.pop() ;
        
            
         if(minstack.size()==0) minnumber=0x7fffffff;
         else{minnumber=minstack.top();}
        
         return c;
        // write your code here
    }

    /*
     * @return: An integer
     */
    int min() {
        if(a.size()==0) return NULL;
        return minnumber;
        // write your code here
    }
};