LeetCode 剑指 Offer 30. 包含min函数的栈

目录结构

1.题目

2.题解

---

1.题目

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

示例：

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.min();   --> 返回 -2.

提示：

1. 各函数的调用总次数不超过 20000 次 

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

辅助栈栈顶始终保存截止目前数据的最小值，当出栈时，若当前出栈元素为目前的最小值，则辅助栈栈顶元素也出栈。

public class Offer30 {

    @Test
    public void test() {
        MinStack minStack = new MinStack();
        minStack.push(-2);
        minStack.push(0);
        minStack.push(-3);
        System.out.println(minStack.min());
        minStack.pop();
        System.out.println(minStack.top());
        System.out.println(minStack.min());
    }

    class MinStack {

        private final Stack<Integer> stack;
        private final Stack<Integer> min_stack;

        public MinStack() {
            stack = new Stack<>();
            min_stack = new Stack<>();
        }

        public void push(int x) {
            stack.push(x);
            if (min_stack.isEmpty() || x <= min_stack.peek()) {
                min_stack.push(x);
            }
        }

        public void pop() {
            if (stack.pop().equals(min_stack.peek())) {
                min_stack.pop();
            }
        }

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

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