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

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 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.
提示：
各函数的调用总次数不超过 20000 次

固定思路

用两个栈
第一个栈stack正常push，pop，top
第二个栈min保存至今的最小值
当为空时直接push当push的值比min.peer()小的话，push当前值
否则push min当前的栈顶值

不知道说的清楚不
不过看完代码就都懂了
有时候代码比描述更好懂

不过效果似乎不咋好

代码

public class MinStack {

    public MinStack() {

    }

    private Stack<Integer> stack = new Stack<>();
    private Stack<Integer> min = new Stack<>();

    public void push(int x) {

        stack.push(x);
        if(min.isEmpty()){
            min.push(x);
        }else if(x<min.peek()){
            min.push(x);
        }else if(x>=min.peek()){
            min.push(min.peek());
        }
    }

    public void pop() {

        if(!stack.isEmpty()){
            stack.pop();
            min.pop();
        }

    }

    public int top() {
        if(!stack.isEmpty())
        return stack.peek();
        return 0;
    }

    public int min() {
        if(!min.isEmpty())
        return min.peek();
        return 0;
    }
}

后来看了一个精选的答案
好似代码没啥区别，但是运行结果区别可大了

代码

class MinStack {


    Stack<Integer> A, B;
    public MinStack() {
        A = new Stack<>();
        B = new Stack<>();
    }
    public void push(int x) {
        A.add(x);
        if(B.empty() || B.peek() >= x)
            B.add(x);
    }
    public void pop() {
        if(A.pop().equals(B.peek()))
            B.pop();
    }
    public int top() {
        return A.peek();
    }
    public int min() {
        return B.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.min();
 */

分析

1. 在内存上的优化
   我的代码在加载类的时候就给stack分配了空间，而大佬的代码是在new实例的时候才分配空间
   而且min栈只有在遇到小于等于栈顶的数时才add

2. 时间上的加速
   可能也是因为min的add变少了，其他的就不知道了