leetcode - Min Stack

leetcode - Min Stack

 

Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.

• push(x) -- Push element x onto stack.
• pop() -- Removes the element on top of the stack.
• top() -- Get the top element.
• getMin() -- Retrieve the minimum element in the stack.

class MinStack {
public:
    int data[100000];
    int minvalue[100000];
    int size=0;
    
    void push(int x) {
        data[size] = x;
        if(size == 0 || x<=minvalue[size-1]){
            minvalue[size] = x;
        }
        else{
            minvalue[size] = minvalue[size-1];
        }
        size++;
    }

    void pop() {
        if(size!=0){
            size--;
        }
        //if(size<=posMin) posMin = -1;
    }

    int top() {
        return data[size-1];
        
    }

    int getMin() {
        if(size==0) return 0;
        return minvalue[size-1];
    }
};

思路： 前几个函数都比较简单，有问题的是getMin，如果每次都遍历找最小的话显然不能达到constan time，所以建立一个和stack大小相同的stack，存放当前最小值，然后每次pop和push时维护。
这样可以常数时间内检索。

空间优化：可以每次push的时候，如果大于当前最小值则不用改变，小于时同时push即可。