剑指offer试题编程练习20(java) 实现取栈中最小元素的函数

 20、定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

思路：实现获取栈中的最小元素的min函数，若只用一个变量来保存当前栈中的最小元素，若该最小元素弹出了则无法知道栈中当前的最小元素是什么了，所以我们借助一个辅助栈来保存当前栈中的最小元素，压栈时若压栈元素小于最小栈最上面的元素则同时压入最小栈，弹出时，若弹出元素等于最小栈最上面的元素则同时把该元素弹出。要利用栈后进先出的特点，画图看会更清晰，更容易理解。

import java.util.Stack;

public class Solution {
    //借助辅助栈来解决问题

    Stack<Integer> stack1 = new Stack<>();
    Stack<Integer> minStack = new Stack<>();    //用来存放最小栈元素
    public void push(int node) {
        //存入一个节点
        stack1.push(node);
        if(minStack.empty()){
            minStack.push(node);
        }else{
            if(node<minStack.peek()){
                minStack.push(node);
            }
        }
    }
    
    public void pop() {
        //弹出一个元素
        if(stack1.peek()==minStack.peek()){
            stack1.pop();
            minStack.pop();
        }else{
            stack1.pop();
        }
    }
    
    public int top() {
        
        return stack1.peek();
        
    }
    
    public int min() {
        //获得栈中的最小元素
        
        return minStack.peek();
    }
}

peek()方法只获取stack的顶端元素，不将其从栈中删除。