剑指offer 面试题21 包含 min 函数的栈

剑指offer 面试题21 包含 min 函数的栈

题目：
定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数，
在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

package algorithm.foroffer.top30;import java.util.Stack;/** * Created by liyazhou on 2017/5/27. * 面试题21：包含 min 函数的栈 * * 题目： *      定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数， *      在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。 * * 问题： *      1. 栈的特性 *      2. 求栈中的最小元素 * * 思路： *      1. 使用索引栈 *      2. 每次添加新元素时，在辅助栈中保存当前数据栈中的最小元素的位置 *//** * 对 T extends Comparable<T> 的理解： * * 1. T 必须实现 Comparable 接口，(并且这个接口类型是T) * 2. 需要给接口的抽象方法 compareTo 的参数类型为 T，也即是 compareTo(T t) * 比如 String implements Comparable<String> *  * @param <T> */class StackWithMin<T extends Comparable<T>>{    Stack<T> dataStack;   // 数据栈，保存压入的数据    Stack<Integer> minIdxStack;  // 索引栈，该位置保存与数据栈的相同位置到栈底之间的最小元素的位置    int minIdx;    public StackWithMin(){        dataStack = new Stack<>();        minIdxStack = new Stack<>();        minIdx = 0;    }    public void push(T element){        if (element == null) throw new RuntimeException("the element is null.");        dataStack.push(element);        if (!minIdxStack.isEmpty()) {            minIdx = dataStack.get(minIdxStack.peek()).compareTo(element) < 0 ? minIdx : dataStack.size()-1;        }        minIdxStack.push(minIdx);    }    public T pop(){        if (dataStack.isEmpty()) throw new RuntimeException("The stack is already null.");        minIdxStack.pop();        return dataStack.pop();    }    public T min(){        if (dataStack.isEmpty()) throw new RuntimeException("The stack is already null.");        return dataStack.get(minIdxStack.peek());    }    @Override    public String toString(){        return dataStack.toString();    }}public class Test21 {    public static void main(String[] args){        StackWithMin<Double> stack = new StackWithMin<>();        stack.push(10.0);        stack.push(15.0);        stack.push(11.0);        stack.push(2.0);        stack.push(1.0);        System.out.println(stack);        System.out.println(stack.pop());        System.out.println(stack);        System.out.println(stack.min());    }}