实现返回栈中最小的元素

引用

    实现一个特殊栈，在实现栈的基本功能的基础上，再实现返回栈中最小的元素。

    要求：1.pop、push、getMin操作的时间复杂度都是O(1)

              2.设计的栈类型可以使用现成的栈结构

    思路：

         1.因为要求时间都是O(1)、所以有可能在删除后会出现找不到最小值问题
         2.创建两个栈，一个data栈，一个min栈，data栈记录压入的数，min栈记录压入数后最小的数
         3.弹出、压入都是同步进行。压入一个data栈元素，那就要和min栈元素进行比较，小的话压入，大的话压入上一个min栈元素，min栈顶元素就是最小元素

          例如：data栈压入3、5、1、7、4、0  min栈记录3、3、1、1、1、0

    代码

public class StackAndQueueConvert {

    private Stack<Integer> stackData; //data栈
    private Stack<Integer> stackMin; //min栈

    /**
     * 初始化栈
     */
    public StackAndQueueConvert(){
        stackData = new Stack<>();
        stackMin = new Stack<>();
    }

    /**
     * 压入操作
     * @param newNum
     */
    public void push(int newNum){
        // min栈没有元素 直接压入
        if(stackMin.isEmpty()) {
            stackMin.push(newNum);
        // 压入元素小于min栈栈顶元素
        }else if(newNum < getMin()){
            stackMin.push(newNum);
        // min栈顶元素比压入元素小
        }else{
            // 获取栈顶元素
            int newMin = stackMin.peek();
            // 继续压入
            stackMin.push(newMin);
        }
        // 把元素压入data栈
        stackData.push(newNum);
    }

    /**
     * 移除栈顶元素
     * @return
     */
    public int pop(){
        if(stackData.isEmpty()){
            throw new RuntimeException("Your stack is empty");
        }
        // 移除min栈栈顶元素
        stackMin.pop();
        // 移除并返回data栈栈顶元素
        return stackData.pop();
    }

    /**
     * 获取最小值
     * @return
     */
    public int getMin(){
        if(stackMin.isEmpty()){
            throw new RuntimeException("Your stack is empty");
        }
        // 返回min栈栈顶元素
        return stackMin.peek();
    }

}

以上就是实现过程，栈结构在数组实现栈结构的博客中讲过，可以大概了解一下实现机制。