队列和栈——最小栈的实现

叨叨猿的青轴日记

于 2025-05-25 16:10:03 发布

阅读量869

点赞数 28

分类专栏：算法那些事文章标签：最小栈队列与栈

本文链接：https://blog.csdn.net/weixin_43192556/article/details/148208987

版权

算法那些事专栏收录该内容

5 篇文章

订阅专栏

01最小栈实现原理

最小栈是一种特殊的栈结构，除了支持标准栈的所有操作外，还需要在O(1)时间内找到栈中的最小元素。

这是一个经典的数据结构设计问题，不通过遍历实现时间复杂度为O(1)，考虑可以用空间换时间的思想。

问题分析

问题描述 LeetCode 155: https://leetcode.cn/problems/min-stack/

设计一个支持以下操作的栈：

- push(x) - 将元素x推入栈中

- pop() - 删除栈顶元素

- top() - 获取栈顶元素

- getMin() - 检索栈中的最小元素

所有操作的时间复杂度都必须是O(1)。

核心思路

使用辅助栈记录最小值：

- data栈：存储实际数据

- min栈：存储到当前位置的最小值

关键点：min栈data栈保持同步，每次push操作，min栈都要压入一个值（当前最小值）。

02基于Java内置Stack

方案一：基于Java内置Stack

实现代码

import java.util.Stack;/** * 基于Java内置Stack的最小栈实现 * 缺点是常数时间慢，有同步开销 */class MinStackWithBuiltIn {    private Stack<Integer> dataStack;   // 存储实际数据    private Stack<Integer> minStack;    // 存储最小值信息        public MinStackWithBuiltIn() {        dataStack = new Stack<>();        minStack = new Stack<>();    }        /**     * 压栈操作     * @param val 要压入的值     */    public void push(int val) {        dataStack.push(val);                // 更新最小值的栈：如果新值更小或相等，或者是第一个元素，则压入栈内        if (minStack.isEmpty() || val <= minStack.peek()) {            minStack.push(val);        } else {            // 保持与dataStack同步数量，所以重复压入当前最小值            minStack.push(minStack.peek());        }    }        /**     * 弹栈操作     */    public void pop() {        dataStack.pop();        minStack.pop();  // 两个栈保持同步    }        /**     * 获取栈顶元素     */    public int top() {        return dataStack.peek();    }        /**     * 获取当前栈中最小元素     */    public int getMin() {        return minStack.peek();    }

时间复杂度：

- push()：O(1) - 栈的基本操作

- pop()：O(1) - 栈的基本操作

- top()：O(1) - 直接访问栈顶

- getMin()：O(1) - 直接访问最小值栈顶

缺点是常数时间慢，即便是O(1)常数时间。

空间复杂度：

- O(n) - 需要两个栈，每个栈最多存n个元素

03基于数组实现的优化

实现代码

/** * 基于数组的最小栈实现 * 优点：常数时间快，无同步开销 * 缺点：需要预估容量上限 */class MinStackWithArray {    private static final int MAXN = 8001;  // 根据面试题题干，确定的最大容量    private int[] dataArray;    // 存实际数据的数组    private int[] minArray;     // 存最小值的数组    private int size;           // 当前栈的大小        public MinStackWithArray() {        dataArray = new int[MAXN];        minArray = new int[MAXN];        size = 0;    }        /**     * 压栈操作     */    public void push(int val) {        dataArray[size] = val;                // 更新最小值数组        if (size == 0 || val <= minArray[size - 1]) {            minArray[size] = val;       // 新的最小值        } else {            minArray[size] = minArray[size - 1];  // 沿用之前的最小值        }        size++;    }        /**     * 弹栈操作     */    public void pop() {        size--;  // 只需要减少size即可，无需清理数据！    }        /**     * 获取栈顶元素     */    public int top() {        return dataArray[size - 1];    }        /**     * 获取当前栈中最小元素     */    public int getMin() {        return minArray[size - 1];    }}