《程序员代码面试指南》——设计一个有getMin功能的栈（Java实现）

题目：

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

要求：

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

思路：用两个栈，一个做正常的pop、push操作，另一个辅助栈用于存放每次操作后的最小值

Java实现：

import java.util.Stack;

public class Main {
    public static void main(String[] args) {
        MyStack stack1 = new MyStack();
        stack1.push(3);
        stack1.push(2);
        stack1.push(1);
        //获取栈顶元素
        int stack = stack1.pop();
        //获取最小值
        int min = stack1.getMin();
        System.out.println(stack + " " + min + " ");

    }
}

class MyStack {
    //satckData用于存放所有的入栈元素
    private Stack<Integer> stackData;
    //stackMin作为辅助栈，用于存放最小值
    private Stack<Integer> stackMin;

    public MyStack(){
        this.stackData=new Stack<Integer>();
        this.stackMin=new Stack<Integer>();
    }

    //入栈
    public void push(int val){
        stackData.push(val);
        //如果stackMin为空或者当前插入的值小于等于stackMin的栈顶元素，就将新的值插入stackMin
        if(stackMin.empty()||val<=stackMin.peek())
            stackMin.push(val);
        //否则插入和当前stackMin栈顶元素一样的值
        else stackMin.push(stackMin.peek());
    }

    //出栈
    public int pop(){
        //先判空
        if(stackData.empty())
            throw new RuntimeException("Your stack is empty.");
        stackMin.pop();
        return stackData.pop();
    }

    //获取最小值，O(1)
    public int getMin(){
        if(stackMin.empty())
            throw new RuntimeException("Your stack is empty.");
        //stackMin的栈顶元素就是最小值
        return stackMin.peek();
    }
}

参考：《程序员代码面试指南》