【Java - J - 30】包含min函数的栈

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。
注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

lc

练习地址
L - 0155】- e - 最小栈⭐⭐⭐

实现

• O(1) -> 增加一个辅助栈使最小元素在栈顶
• 最小元素出栈后，次小元素的保存 -> 入栈时，入栈元素和最小值比较，将新的最小值压入辅助栈
• 出栈操作 -> 两个栈都执行

例子：
入栈顺序：3 4 2 1

步骤	操作	数据栈	辅助栈	最小值
1	压入3	3	3	3
2	压入4	3,4	3,3	3
3	压入2	3,4,2	3,3,2	2
4	压入1	3,4,2,1	3,3,2,1	1
5	弹出	3,4,2	3,3,2	2
6	弹出	3,4	3,3	3
7	压入0	3,4,0	3,3,0	0

注：没有重复把最小值push，但最小值pop的时候做了equals的判断，避免了唯一重复最小值出栈，效果一样
B.peek() >= x 避免了重复最小值被弹出。

    class MinStack {
        Stack<Integer> A, B;

        public MinStack() {
            A = new Stack<>();
            B = new Stack<>(); // 辅助栈
        }

        public void push(int x) {
            A.push(x);                  //= 一定要有
            if (B.empty() || B.peek() >= x) { //B.peek() >= x 避免了重复最小值被弹出。
                B.push(x);
            }
        }

        public void pop() {
            if (A.pop().equals(B.peek())) {
                B.pop();
            }
        }

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

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