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

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

示例:

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

思路

思路一：用栈实现：

1. 先建立两个栈A、B，A正常使用，B用来存储最小值；
2. 在往A中插入元素的同时，判断B是否为空或者B的栈顶元素是否小于这个新插入的元素，如果B为空或者B的栈顶元素大于新插入的元素，就将这个元素同时插入的B中；
3. 在删除A的栈顶元素时，判断B的栈顶元素是否等于这个要删除的元素，如果等于，那么B的这个栈顶元素也要删除。因为A 的这个元素删除了，之后所进行的找最小值的操作，都是在删除这个栈顶元素后的栈中寻找的，不包括要删除的这个栈顶元素，所以B的这个和A相同的栈顶元素也要删除。
4. 经过上面的操作，B的栈顶元素就是最小值，所以找最小值直接取B的栈顶元素即可

思路二：用数组实现：

1. 用数组元素实现相对简单。先建立一个较大的数组（也可在下面进行扩容操作），数组的长度为0。
2. 每插入一个元素，就是将该元素赋给数组的当前长度，并且数组的长度+1。
3. 每 删除一个元素，就是使数组的长度-1。
4. 根据先进后出，取栈顶元素，就相当于取数组的最后一个元素。
5. 取最小值则是将数组遍历比较即可。

代码

法一：用栈实现

class MinStack {
    Stack<Integer> A = new Stack<>();
    Stack<Integer> B = new Stack<>();//记录最小元素的栈

    /** initialize your data structure here. */
    public MinStack() {

    }
    
    public void push(int x) {
        //判断插入的元素是否为最小的值
        A.add(x);
        
        //此处一定得是大于等于，因为他可能会插入多个相同的值，如果只按一个算，会造成溢出
        if(B.isEmpty() || B.peek() >= x){
            B.add(x);
        }
    }
    
    public void pop() {
        //判断栈顶元素是不是最小的
        if(B.peek().equals(A.pop())){
            B.pop();
        }
    }
    
    public int top() {
        return A.peek();
    }
    
    public int min() {
        return B.peek();
    }
}

法二：用数组实现

class MinStack {
    private int[] data = new int[10000];
    int size = 0;

    /** initialize your data structure here. */
    public MinStack() {

    }
    
    public void push(int x) {
        data[size] = x;
        size++;
    }
    
    public void pop() {
        if(size == 0){
            return;
        }
        size--;
    }
    
    public Integer top() {
        if(size == 0){
            return null;
        }
        return data[size - 1];
    }
    
    public Integer min() {
        if(size == 0){
            return null;
        }
        int ret = data[0];
        for(int i = 1;i < size;i++){
            if(data[i] < ret){
                ret = data[i];
            }
        }
        return ret;
    }
}

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/bao-han-minhan-shu-de-zhan-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。