栈的最小值，进出栈取最小值全部是O（1）

最新推荐文章于 2024-09-14 18:46:05 发布

shilong233

最新推荐文章于 2024-09-14 18:46:05 发布

阅读量39

点赞数

文章标签： java c++ 开发语言

本文链接：https://blog.csdn.net/shilong233/article/details/129734438

版权

该文章介绍了一个名为MinStack的数据结构实现，通过维护两个栈a和b，栈b作为辅助栈来存储当前的最小值。当元素入栈时，如果它小于或等于b栈顶元素，则将其同时压入a和b栈；否则只压入a栈。出栈时，如果a栈顶元素等于b栈顶元素，表示它是最小值，先弹出b栈顶元素。这样，b栈始终保持最小值。getMin方法直接返回b栈顶元素即可获取最小值。

摘要由CSDN通过智能技术生成

一定要注意，进站之后要用equals判断，用==判断出来，栈是两个不同的对象进站

class MinStack {
        //新建两个栈，一个a，一个b，b用来辅助a的存储
        Stack<Integer> a = new Stack<>();
        Stack<Integer> b = new Stack<>();
        /** initialize your data structure here. */
        public MinStack() {

        }

        public void push(int x) {
            //进站时，如果b中为空则直接进入辅助栈，如果不为空就判断当前存储元素是否比辅助b栈顶元素小
            if (b.isEmpty()) {
                b.push(x);
                a.push(x);
            }else {
                //当前元素比栈顶小，就进入辅助栈，反制则直接进入a
                a.push(x);
                if (x <= b.peek())
                    b.push(x);
            }

        }

        public void pop() {
            //出栈时，判断当前元素是否为辅助b的栈顶
            if (a.peek().equals(b.peek())){
                //如果是则弹出b
                b.pop();
            }
            a.pop();
        }

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

        public int getMin() {
            return b.peek();
        }
    }