剑指 Offer 30. 包含min函数的栈

思路

由于需要在常数时间内找到最小的元素，那么说明肯定是不能使用遍历，因为遍历是 O(n) 级别的时间，那么只能使用辅助空间进行存储，这是一种空间换时间的思想。

这里我们设置两个栈：普通栈和辅助栈。

解法：两个栈实现

class MinStack {

    Stack<Integer> stack;
    Stack<Integer> minStack;
    /** initialize your data structure here. */
    public MinStack() {
        stack = new Stack<>();
        minStack =  new Stack<>();
    }
    
    public void push(int x) {
        stack.push(x);

        // 最小栈为null直接压栈
        if(minStack.empty()){
            minStack.push(x);
        }else{
            // 不为Null则和栈顶元素进行判断
            if(minStack.peek()>=x){
                minStack.push(x);
            }else{
                // 压栈当前最小栈的栈顶元素
                minStack.push(minStack.peek());
            }
        }
    }
    
    public void pop() {
        stack.pop();
        minStack.pop();
    }
    
    public int top() {
        return stack.peek();
    }
    
    public int min() {
        return minStack.peek();
    }
}

参考:
1、https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof/solution/mian-shi-ti-30-bao-han-minhan-shu-de-zhan-fu-zhu-z/
2、https://blog.algomooc.com/030.html#%E4%B8%89%E3%80%81%E5%8F%82%E8%80%83%E4%BB%A3%E7%A0%81