Java语言设计一个有getMin功能的栈

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

要求：

• pop、push、getMin操作的时间复杂度都是O(1)
• 设计的栈类型可以使用现成的栈结构

实现代码：

public class GetMinStack {
        public static class MyStack1 {
            /**
             * 两个栈，其中stacMin负责将最小值放在栈顶，stackData通过获取stackMin的peek()函数来获取到栈中的最小值
             */
            private Stack<Integer> stackData;
            private Stack<Integer> stackMin;

            /**
             * 在构造函数里面初始化两个栈
             */
            public MyStack1() {
                stackData = new Stack<Integer>();
                stackMin = new Stack<Integer>();
            }

            /**
             * 该函数是stackData弹出栈顶数据，如果弹出的数据恰好等于stackMin的数据，那么stackMin也弹出
             *
             * @return
             */
            public Integer pop() {
                Integer num = (Integer) stackData.pop();
                if (num == getmin()) {
                    return (Integer) stackMin.pop();
                }
                return num ;
            }

            /**
             * 该函数是先判断stackMin是否为空，如果为空，就push新的数据，如果这个数小于stackMin中的栈顶元素，那么stackMin需要push新的数，不管怎么样
             * stackData都需要push新的数据
             *
             * @param value
             */
            public void push(Integer value) {
                if (stackMin.isEmpty()) {
                    stackMin.push(value);
                } else if (value < getmin()) {
                    stackMin.push(value);
                }
                stackData.push(value);
            }

            /**
             * 该函数是当stackMin为空的话第一次也得push到stackMin的栈中，返回stackMin的栈顶元素
             *
             * @return
             */
            public Integer getmin() {
                if (stackMin == null) {
                    throw new RuntimeException("stackMin is empty");
                }
                return (Integer) stackMin.peek();
            }
        }

        public static void main(String[] args) throws Exception {
            /**
             * 要注意要将MyStack1声明成静态的，静态内部类不持有外部类的引用
             */
            MyStack1 stack1 = new MyStack1();
            stack1.push(3);
            System.out.println(stack1.getmin());
            stack1.push(4);
            System.out.println(stack1.getmin());
            stack1.push(1);
            System.out.println(stack1.getmin());
            System.out.println(stack1.pop());
            System.out.println(stack1.getmin());

            System.out.println("=============");
        }
    }