简洁写法剑指 Offer 30. 包含min函数的栈

Posted 2021-06-30 来老铁干了这碗代码

立志用最少的代码做最高效的表达

---

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 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.

提示：
各函数的调用总次数不超过 20000 次

---

class MinStack {
    Stack<Integer>s;
    Stack<Integer>min;
    public MinStack() {
        s = new Stack<>();
        min = new Stack<>();
    }

    public void push(int x) {
        s.push(x);
        if(min.isEmpty()) min.push(x);
        else {
            if(min.peek() > x) min.push(x);
            else min.push(min.peek());
        }
    }

    public void pop() {
        s.pop(); min.pop();
    }

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

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

---

完整可运行代码

import java.util.LinkedList;
import java.util.Queue;
import java.util.Stack;

public class 剑指Offer30_包含min函数的栈 {
    class MinStack {

        Stack<Integer>s = new Stack<>();
        Stack<Integer>min = new Stack<>();
        public MinStack() {

        }

        public void push(int x) {
            s.add(x);
            if(min.isEmpty()) min.push(x);
            else {
                if(min.peek() > x) min.add(x);
                else min.add(min.peek());
            }
        }

        public void pop() {
            if(!s.isEmpty()) {
                s.pop(); min.pop();
            }
        }

        public int top() {
            if(!s.isEmpty()) return s.peek();
            else return -1;
        }

        public int min() {
            if(!min.isEmpty()) return min.peek();
            else return -1;
        }
    }

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

---

天若有情天亦老，人间正道是沧桑。