实现带有返回栈中最小元素功能的栈结构

实现一个特殊的栈，在实现栈的基本功能的基础上，再实现返
回栈中最小元素的操作
要求：pop、push、getMin操作的时间复杂度都是O(1)

版本1

package com.chanmufeng.codingInterviewGuide;

import java.util.Stack;

public class GetMinStack1 {
    private static Stack<Integer> dataStack = new Stack<>();
    private static Stack<Integer> minStack = new Stack<>();

    public static void push(int newNum) {
        dataStack.push(newNum);
        if (minStack.isEmpty()) {
            minStack.push(newNum);
        } else {
            minStack.push(Math.min(minStack.peek(), newNum));
        }
    }

    public static int pop() {
        if (dataStack.isEmpty()) {
            throw new RuntimeException("stack is empty");
        }
        minStack.pop();
        return dataStack.pop();
    }

    public static int getMin() {
        if (minStack.isEmpty()) {
            throw new RuntimeException("stack is empty");
        }
        return minStack.peek();
    }

    public int size() {
        return dataStack.size();
    }


    public static void main(String[] args) {
        GetMinStack1 stack = new GetMinStack1();
        stack.push(3);
        stack.push(4);
        stack.push(5);
        stack.push(1);
        stack.push(2);
        stack.push(1);

        while (stack.size()>0){
            System.out.println(stack.getMin());
            stack.pop();
        }
    }
}

版本2

package com.chanmufeng.codingInterviewGuide;

import java.util.ArrayList;
import java.util.Stack;

/**
 * 实现含有获取栈内最小元素功能的栈结构
 * 版本二，优化minStack空间复杂度
 */
public class GetMinStack2 {
    private static ArrayList<Integer> dataStack = new ArrayList<>();
    private static Stack<Integer> minStack = new Stack<>();

    public static void push(int newNum) {
        dataStack.add(newNum);
        if (minStack.isEmpty() || getMin() > newNum) {
            minStack.push(dataStack.size() - 1);
        }
    }

    public static int pop() {
        if (dataStack.isEmpty()) {
            throw new RuntimeException("stack is empty");
        }
        int size = dataStack.size();
        int res = dataStack.remove(--size);

        //如果弹出的元素的索引为minStack的顶部元素，则出栈
        if (minStack.peek() == size) {
            minStack.pop();
        }
        return res;
    }

    public static int getMin() {
        if (minStack.isEmpty()) {
            throw new RuntimeException("stack is empty");
        }
        return dataStack.get(minStack.peek());
    }

    public int size() {
        return dataStack.size();
    }


    public static void main(String[] args) {
        GetMinStack2 stack = new GetMinStack2();
        stack.push(3);
        stack.push(4);
        stack.push(5);
        stack.push(1);
        stack.push(2);
        stack.push(1);

        while (stack.size() > 0) {
            System.out.println(stack.getMin());
            stack.pop();
        }
    }
}