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.复制代码

刚看到题目。。本身第一个想法是要不要开一个链表慢慢比较栈中数据的大小。。后来百度一下，看到了正确思路。

首先开两个栈，一个栈a（放元素x），一个栈b（放最小值）。每当有元素x进栈时，元素x先跟栈b的栈顶比较，若是元素x比栈顶元素小，元素x入栈a、b；若是元素x大于栈顶元素，元素x入栈a。出栈时，出栈元素跟栈b的栈顶比较，若是出栈元素与栈b栈顶相等，栈b.pop()。这种方法可以避免，若是当前栈的最小值出栈时，就没法找到栈的最小值。

代码实现：

链表实现：

class MinStack {
    private List<Integer> data;
    private List<Integer> min;
    /** initialize your data structure here. */
    public MinStack() {
        data = new ArrayList<>();
        min = new ArrayList<>();
    }
    
    public void push(int x) {
        if(data.isEmpty() || x <= min.get(min.size()-1)) {
            min.add(x);
        }
        data.add(x);
    }
    
    public void pop() {
        if(data.get(data.size()-1).equals(min.get(min.size()-1))){
            min.remove(min.size()-1);
        }
        data.remove(data.size()-1);
    }
    
    public int top() {
        return data.get(data.size()-1);
    }
    
    public int getMin() {
        return min.get(min.size()-1);
    }
}复制代码

栈实现：

class MinStack {
    private Stack<Integer> data;
    private Stack<Integer> min;
    /** initialize your data structure here. */
    public MinStack() {
        data = new Stack();
        min = new Stack();
    }
    
    public void push(int x) {
        if(data.isEmpty() || x <= min.peek()) {
            min.push(x);
        }
        data.push(x);
    }
    
    public void pop() {
        if(data.peek().equals(min.peek())){
            min.pop();
        }
        data.pop();
    }
    
    public int top() {
        return data.peek();
    }
    
    public int getMin() {
        return min.peek();
    }
}
复制代码