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

2.2.1题目

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 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.

2.2.2解法：辅助栈

难点：min的时间复杂度为O(1)

思路：

可通过建立辅助栈实现；

• 数据栈 dataStack： 栈 dataStack用于存储所有元素，保证入栈 push() 函数、出栈 pop() 函数、获取栈顶 top() 函数的正常逻辑。
• 辅助栈 minStack： 栈 minStack中存储栈 dataStack中所有 非严格降序 的元素，则栈 dataStack中的最小元素始终对应栈 minStack的栈顶元素，即 min() 函数只需返回栈 minStack的栈顶元素即可。

class MinStack {

    private Stack<Integer> dataStack;   //数据栈
    private Stack<Integer> minStack;    //最小值栈

    public MinStack() {
     dataStack = new Stack<Integer>(); 
     minStack = new Stack<Integer>();  
    }
    
    public void push(int x) {
        dataStack.push(x);  //数据压入数据栈
        //假如最小值栈为空，或者最小值栈的栈顶大于当前x值  x压入最小值栈
        if(minStack.empty() || minStack.peek() >= x){   
            minStack.push(x);
        }
    }
    
    public void pop() {
        //注意dataStack.pop()即使再if里面也会弹栈
        //如果数据栈弹出的数据等于最小值栈的栈顶栈  则最小值栈也弹栈
        if(dataStack.pop().equals(minStack.peek())){
            minStack.pop();
        }
    }
    
    public int top() {
        return dataStack.peek();
    }
    
    public int min() {
        return minStack.peek();
    }
}