LeetCode 初级 - 最小栈

最小栈

设计一个支持 push，pop，top 操作，并能在常数时间内检索到最小元素的栈。

• push(x) – 将元素 x 推入栈中。
• pop() – 删除栈顶的元素。
• top() – 获取栈顶元素。
• getMin() – 检索栈中的最小元素。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.getMin();   --> 返回 -2.

分析

对于普通的栈而言，我们基本不需要额外实现什么，但因为常数时间内检索到最小元素的栈的要求，我们需要记录最小元素。同时要考虑到这个最小元素是有可能被退栈的。因此我们仍需要一个额外的栈来记录最小元素。

代码

  public class MinStack {
        private Stack<Integer> stack;
        private Stack<Integer> minStack;

        public MinStack() {
            stack = new Stack<Integer>();
            minStack = new Stack<Integer>();
        }

        public void push(int number) {
            stack.push(number);
            if (minStack.empty() || minStack.peek() >= number)
                minStack.push(number);
        }

        public int pop() {
            if (stack.peek().equals(minStack.peek()) )
                minStack.pop();
            return stack.pop();
        }
        public int top(){
            return stack.peek();
        }

        public int getMin() {
            return minStack.peek();
        }
    }
    /**
     * Your MinStack object will be instantiated and called as such:
     * MinStack obj = new MinStack();
     * obj.push(x);
     * obj.pop();
     * int param_3 = obj.top();
     * int param_4 = obj.getMin();
     */