【剑指offer】面试题30：包含min函数的栈

完整代码地址

完整代码地址

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。

思路

用一个值来保存最小值是行不通的，因为栈pop操作后，最小值就变了。
但是可以使用另外一个栈minStack来保存最小值。
每次push操作时，将当前最小值与新元素对比得出新的最小值，同时push进minStack。
每次pop操作时，两个栈同时pop。

代码

public class _30_MinInStack {

    private Stack<Integer> stack = new Stack<>();
    private Stack<Integer> minStack = new Stack<>();

    public void push(int node) {
        stack.push(node);
        if(minStack.isEmpty()) {
            minStack.push(node);
        }
        else if(minStack.peek() > node) {
            minStack.push(node);
        }
        else {
            minStack.push(minStack.peek());
        }
    }

    public void pop() {
        stack.pop();
        minStack.pop();
    }

    public int top() {
        return stack.peek();
    }

    public int min() {
        return minStack.peek();
    }
}

测试

public class _30_Test {

    public static void main(String[] args) {
        test1();
    }

    /**
     * 1.新压入栈的数字比之前的最小值大
     * 2.新压入栈的数字比之前的最小值小
     * 3.弹出栈的数字不是最小元素
     * 4.弹出栈的数字是最小元素
     */
    private static void test1() {
        _30_MinInStack mis = new _30_MinInStack();
        mis.push(4);
        mis.push(5);
        mis.push(6);
        MyTest.equal(mis.top(), 6);
        MyTest.equal(mis.min(), 4);

        mis.pop();
        mis.pop();
        MyTest.equal(mis.top(), 4);
        MyTest.equal(mis.min(), 4);

        mis.push(2);
        MyTest.equal(mis.top(), 2);
        MyTest.equal(mis.min(), 2);

        mis.pop();
        MyTest.equal(mis.top(), 4);
        MyTest.equal(mis.min(), 4);
    }
}