Leetcode 155. 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 stac

题意理解

该题是要设计一个栈，实现push(x),pop(),top(),getMin()操作。
这里有个关键点实现getMin()。要想获取当前栈的最小值，我们就得创建另外一个栈来记录每个点的最小值。当然这是一种实现的方法。
那么有什么更好的方法呢。为了使用一个栈来记录最小值，我们就得让栈记录更多的信息。例如，在这里，用栈中元素的正负来记录栈的状态。如果删除的是最小值，那么我们通过记录的负数恢复到原来的值。如果删除的不是最小值，那么还保持不变。

代码

class MinStack {
    long min;
    Stack<Long> stack;

    //最小栈初始化函数
    public MinStack() {
        stack = new Stack<Long>();
    }
    //最小栈进栈，push进x与当前最小值的差
    //小于零表示，在这里需要改变最小值，大于零就是正常
    public void push(int x) {
        if(stack.isEmpty())
        {
            min = x;
            stack.push(0L);
            return;
        }
        stack.push(x - min);
        if(x < min)  min = x;
    }
    //弹栈直接弹出栈顶元素，如果它是负的，就修改最小值
    public void pop() {
        long pop = stack.pop();
        if( pop < 0 ) min = min - pop; 
    }
    //栈顶元素获取，大于零表示是当前最小值的差值，小于零等于零表示当前栈顶是最小值
    //并且弹出的时，最小值将变成min - top
    public int top() {
        long top=stack.peek();
        if (top>0){
            return (int)(top+min);
        }else{
           return (int)(min);
        }
    }

    public int getMin() {
        return (int ) min;
    }
}