剑指Offer——带min()函数的栈

题目描述：实现一个可以在O（1）时间内找到最小值的栈，即带min()函数的栈

分析：因为需要在O（1）时间内得到栈的最小值，因此我们很容易想到要用一个容器来保存当前栈的最小值，但是如果当元素弹出的时候最小值弹出了怎么办呢？那么说明我们需要保存上一个最小值，那么我们想到，如果能够使用另一个栈，来保存每一次压栈后的最小值，那么当栈顶弹出后，最小栈的栈顶是上一次压栈的最小值。

例如上图：首先压栈1，此时最小值为1，压入最小栈；然后压栈2，此时最小值为1，压入最小栈；然后压栈0，此时最小值为0，压入最小栈；

那么说明每次压栈的时候，都要把当前元素与最小栈栈顶的元素比较，把较小的元素压入最小栈，在出栈的时候，把最小栈的栈顶一起弹出。

实现代码

import java.util.Stack;

public class Solution {
    //思路：要能够在O(1)时间内得到min值，那么需要使用2个栈来实现，一个栈用于保存元素，
    //另一个栈用于保存当前元素栈的最小值
    Stack<Integer> s1 = new Stack<>();//用于保存元素
    Stack<Integer> s2 = new Stack<>();//用于保存最小值
    
    public void push(int node) {
        s1.push(node);
        if(s2.isEmpty()){
            s2.push(node);
        }
        else{
            int lastMin = s2.peek();
            if(node<lastMin){
                s2.push(node);
            }
            else{
                s2.push(lastMin);
            }
        }
    }
    
    public void pop() {
        s1.pop();
        s2.pop();
    }
    
    public int top() {
        return s1.peek();
    }
    
    public int min() {
        return s2.peek();
    }
}