[剑指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.

Java实现：

• 本来打算用一个实时的min变量来记录当下最小值，后来发现如果最小值正好被取出的时候，没有办法找到上一个最小值。
• 可以用一个辅助栈来实现最小值的实时存储
• 在push中，向主栈中直接加入元素，在辅助栈中存入当前最小值，如果x不小于当前最小值，则辅助栈存入栈顶元素（即当前最小值再存一遍）

class MinStack {
    private Stack<Integer> stack;
    private Stack<Integer> stack_min;
    /** initialize your data structure here. */
    public MinStack() {
        stack = new Stack<>();
        stack_min = new Stack<>();
        stack_min.push(Integer.MAX_VALUE);
    }

    public void push(int x) {
        stack.push(x);
        if(x<stack_min.peek()){
            stack_min.push(x);
        }else stack_min.push(stack_min.peek());
    }

    public void pop() {
        stack.pop();
        stack_min.pop();
    }

    public int top() {
        return stack.peek();
    }

    public int min() {
        return stack_min.peek();
    }
}

 官方解法

同样用一个主栈A存储所有元素，保证保证入栈 push() 函数、出栈 pop() 函数、获取栈顶 top() 函数的正常逻辑。

辅助栈B中存储A中所以非严格降序元素的子序列则栈 A 中的最小元素始终对应栈 B 的栈顶元素。此时， min() 函数只需返回栈 B 的栈顶元素即可。

1. push()

• 元素x压入栈A
• B为空或者x<栈B的栈顶元素，执行元素x压入栈B

2. pop()

• 执行栈A元素出栈，将出栈元素记为y
• 如果y等于B的栈顶元，执行栈B元素出栈

3. top()

• 返回A的栈顶元素

4. min() 

• 返回B的栈顶元素

Java实现

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();
    }
}