包含min函数的栈 java 实现

最新推荐文章于 2026-06-24 09:51:58 发布

原创最新推荐文章于 2026-06-24 09:51:58 发布 · 278 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法 #java

剑指offer 专栏收录该内容

31 篇文章

订阅专栏

本文介绍了如何在Java中实现一个数据结构栈，其中包含一个min函数，使得push、pop和获取最小值的时间复杂度均为O(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.

LeetCode ：包含 min 函数的栈

解题思路

定义两个栈，dataStack 用于存储数据；minStack 用于存储数据栈中所含有数据的最小元素。 minStack 栈顶元素为当前栈中最小的值；minStack 中元素个数始终与 dataStack 的元素个数相同。

在对栈 dataStack 进行 push 入栈操作时，同时对 minStack 进行操作：当 minStack 中为空时直接压入栈中，当加入的元素比 minStack 中栈顶元素小时，将新加入的元素压入 minStack 中; 否则将 minStack 中的栈顶元素压入栈中。从而使 minStack 栈顶元素一直为当前栈中最小的值。
在对栈 dataStack 进行 pop 出栈操作时，同时对 minStack 进行出栈操作。
包含min函数的栈

代码

class MinStack {
		// 定义两个栈，dataStack 用于存储数据；minStack 用于存储数据栈中所含有数据的最小元素
        private Stack<Integer> dataStack = null;
        private Stack<Integer> minStack = null;

        /** initialize your data structure here. */
        public MinStack() {
        // 初始化两个栈
            dataStack = new Stack<Integer>();
            minStack = new Stack<Integer>();

        }

        public void push(int x) {
            dataStack.push(x);
            // dataStack 中加入一个新的数据时，minStack中为空时直接压入栈中，当加入的元素比 minStack 
            // 中栈顶元素小时，将新加入的元素压入 minStack 中; 否则将 minStack 中的栈顶元素压入栈中。
            if (minStack.isEmpty() || minStack.peek() > x) {
                minStack.push(x);
            } else {
                minStack.push(minStack.peek());
            }
        }

        public void pop() {
            dataStack.pop();
            minStack.pop();
        }
//        返回 dataStack 栈顶元素
        public int top() {
            return dataStack.peek();
        }

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