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

题目

力扣链接——剑指 Offer 30. 包含min函数的栈

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

class MinStack {

    /** initialize your data structure here. */
    public MinStack() {

    }
    
    public void push(int x) {

    }
    
    public void pop() {

    }
    
    public int top() {

    }
    
    public int min() {

    }
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.min();
 */

做法1:

栈中不仅保存元素，还同时保存当前最小值。定义一个数据结构Node来保证栈可以同时保存这两种数据。

class MinStack {
    class Node {
        int min;
        int value;
        public Node(int min, int value) {
            this.min = min;
            this.value = value;
        }
    }

    LinkedList<Node> stack = new LinkedList<>();
    
    public MinStack() {}
    
    public void push(int x) {
        Node node = stack.peek();
        int minValue = x;
        if (node != null && node.min < x) {
            minValue = node.min;
        }

        stack.push(new Node(minValue, x));
        
    }
    
    public void pop() {
        stack.pop();
    }
    
    public int top() {
        return stack.peek().value;
    }
    
    public int min() {
        return stack.peek().min;
    }
}

方式2

每次都保存点当前最小值可能会导致数据重复，这个当前最小值可以另存在另一个栈中，当且仅当最小是更新时压入新的最小值

class MinStack {

    Stack<Integer> stack1 = new Stack<>();
    Stack<Integer> stack2 = new Stack<>();

    public MinStack() {}
    
    public void push(int x) {
        stack1.push(x);
        if (stack2.isEmpty() || stack2.peek() >= x) {
            stack2.push(x);
        }
    }
    
    public void pop() {
        if(stack1.pop().equals(stack2.peek())) {
            stack2.pop();
        }
    }
    public int top() {
        return stack1.peek();
    }
    public int min() {
        return stack2.peek();
    }

}