Hot 100 --- 最小栈

本文概览:本文讲解最小栈的核心思路:用辅助栈记录每个时刻的最小值,实现 O(1) 获取最小值


一、题目


二、题目分析

1. 题目要求

题目要求设计一个支持以下操作的栈:

  • push(x):将元素 x 推入栈中
  • pop():删除栈顶元素
  • top():获取栈顶元素
  • getMin():获取栈中的最小元素

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

2. 难点分析

前三个操作都是栈的基本操作,直接用 Java 的 ArrayDequeLinkedList 就能实现。

难点在于 getMin()

最直觉的做法 :用一个 int min 来记录最小值,每次 push 时更新 min = Math.min(min, value)

看起来没问题

复制代码
push(3)  → 栈:[3],min = 3
push(1)  → 栈:[3, 1],min = min(3, 1) = 1
push(2)  → 栈:[3, 1, 2],min = min(1, 2) = 1

问题出在 pop()

pop() 删除的元素不是最小值时,min 不受影响,没问题:

复制代码
pop()    → 删除 2,栈:[3, 1],min = 1 ✓(2 不是最小值,min 不变)

但当 pop() 删除的恰好是最小值时,min 就失效了:

复制代码
pop()    → 删除 1,栈:[3],min = ?(1 是最小值,被删了,新的最小值是多少?)

此时 min 变量里存的还是 1,但 1 已经被弹出栈了。要找到新的最小值,只能重新遍历整个栈,时间复杂度变成 O(n)。

根本原因 :一个 min 变量只能记住"当前"的最小值,但无法记住"历史"的最小值。当最小值被弹出后,我们不知道"上一个最小值"是什么------因为这个信息已经被覆盖了。

例子

复制代码
push(5)  → min = 5
push(3)  → min = 3(覆盖了 5)
push(1)  → min = 1(覆盖了 3)
pop()    → 删除 1,min = ?(3 被覆盖了,找不回来了)

所以我们需要一种方式,能够记住每一个时刻的最小值,而不是只记住当前这一个。

3. 核心思路:辅助栈

关键观察 :我们需要记录每个时刻的栈的最小值,而不是只记录一个全局最小值。

解决方案 :用一个辅助栈 minStack,和主栈同步操作:

  • push 时,往 minStack 也压入当前的最小值
  • pop 时,minStack 也出栈
  • getMin 时,直接返回 minStack 的栈顶

这样主栈的每个时刻的最小值,都和辅助栈的栈顶一一对应。


三、思路概览

java 复制代码
class MinStack {
    private final Deque<Integer> stack;
    private final Deque<Integer> minStack;

    public MinStack() {
        stack = new LinkedList<>();
        minStack = new LinkedList<>();
    }
    
    public void push(int value) {
        stack.push(value);
        if (minStack.isEmpty()) {
            minStack.push(value);
        }
        else {
            minStack.push(Math.min(value, minStack.peek()));
        }
    }
    
    public void pop() {
        if (stack.isEmpty()) {
            return;
        }
        int top = stack.pop();
        minStack.pop();
    }
    
    public int top() {
       if (stack.isEmpty()) {
           return -1;
       }
       return stack.peek();
    }
    
    public int getMin() {
       if (minStack.isEmpty()) {
              return -1;
         }
       return minStack.peek();
    }
}

思路简要说明:

  1. 两个栈同步操作 :主栈 stack 存储数据,辅助栈 minStack 存储每个时刻的最小值
  2. push 时minStack 压入 Math.min(value, minStack.peek()),即当前值和之前最小值的较小者
  3. pop 时:两个栈同时出栈,保持同步
  4. getMin 时 :直接返回 minStack.peek(),O(1) 时间复杂度

四、思路详解

第一步:为什么不能用一个 int 记录最小值?

如果用一个变量 min 记录最小值:

复制代码
push(3)  → min = 3
push(1)  → min = 1
pop()    → 删除 1,min = ?(需要遍历栈才能知道是 3)

问题pop() 删除最小值后,无法 O(1) 获取新的最小值。

第二步:辅助栈的思路

核心思想:记录每个时刻的最小值,而不是只记录一个全局最小值。

辅助栈 minStack 的栈顶始终是当前主栈的最小值。

push 时

java 复制代码
minStack.push(Math.min(value, minStack.peek()));
  • 如果 minStack 为空,直接压入 value
  • 否则,压入 valueminStack.peek() 的较小值

pop 时

java 复制代码
minStack.pop();

两个栈同时出栈,保持同步。

getMin 时

java 复制代码
return minStack.peek();

直接返回辅助栈的栈顶,就是当前最小值。

第三步:为什么这样是对的?

关键:辅助栈的每个位置,记录的是主栈从栈底到该位置的最小值。

比如主栈是 [3, 1, 2]

  • 栈底是 3,此时最小值是 3
  • 加入 1,此时最小值是 min(3, 1) = 1
  • 加入 2,此时最小值是 min(1, 2) = 1

辅助栈就是 [3, 1, 1],栈顶 1 就是当前最小值。

pop() 删除 2 时,辅助栈也出栈,变成 [3, 1],栈顶 1 还是最小值。

当再次 pop() 删除 1 时,辅助栈也出栈,变成 [3],栈顶 3 就是新的最小值。

完整执行过程

push(3), push(1), push(2), pop(), getMin(), pop(), getMin() 为例:

复制代码
初始:stack = [], minStack = []

push(3):
  stack.push(3)       → stack = [3]
  minStack 为空,压入 3 → minStack = [3]

push(1):
  stack.push(1)       → stack = [3, 1]
  min(1, 3) = 1,压入 1 → minStack = [3, 1]

push(2):
  stack.push(2)       → stack = [3, 1, 2]
  min(2, 1) = 1,压入 1 → minStack = [3, 1, 1]

pop():
  stack.pop() = 2     → stack = [3, 1]
  minStack.pop() = 1  → minStack = [3, 1]

getMin():
  return minStack.peek() = 1 ✓

pop():
  stack.pop() = 1     → stack = [3]
  minStack.pop() = 1  → minStack = [3]

getMin():
  return minStack.peek() = 3 ✓

再以 push(5), push(3), push(4), pop(), getMin() 为例:

复制代码
初始:stack = [], minStack = []

push(5):
  stack = [5]
  minStack = [5]

push(3):
  stack = [5, 3]
  min(3, 5) = 3
  minStack = [5, 3]

push(4):
  stack = [5, 3, 4]
  min(4, 3) = 3
  minStack = [5, 3, 3]

pop():
  stack = [5, 3]
  minStack = [5, 3]

getMin():
  return minStack.peek() = 3 ✓

五、总结

最小栈的核心思路:

  1. 辅助栈记录每个时刻的最小值:而不是只记录一个全局最小值
  2. push 时 :辅助栈压入 Math.min(value, minStack.peek())
  3. pop 时:两个栈同时出栈,保持同步
  4. getMin 时:直接返回辅助栈的栈顶,O(1) 时间复杂度
  5. 空间换时间:辅助栈占用 O(n) 空间,但所有操作都是 O(1)

时间复杂度:所有操作 O(1)

空间复杂度:O(n),辅助栈最多存储 n 个元素

相关推荐
疯狂打码的少年3 小时前
【数据结构】栈的应用:表达式求值(后缀表达式)
java·数据结构·笔记·算法
szephyr3 小时前
腾讯云 ADP 智能体的 Skills 版本回滚总是回到旧配置,是缓存没清还是版本管理没开?
java·缓存·腾讯云
云烟成雨TD3 小时前
Micrometer 系列【39】链路追踪:入门案例 | 环境准备
java·云原生·链路追踪
HiDev_3 小时前
【非标自动化】plc执行顺序问题、扫描周期问题
深度学习·算法·机器学习
chuan.bai3 小时前
Java RAG 实战(第 3 篇):从交互式聊天到多轮上下文
java·人工智能·macos·ai
952364 小时前
Sentinel
java·后端·spring·sentinel·springcloud
阿pin4 小时前
Java随笔-JDK7 HashMap头插法为何能导致死循环?
java·开发语言·hashmap
高洁014 小时前
工信部教考中心证书-人工智能系列本系列
人工智能·python·深度学习·算法
陈王卜4 小时前
Reduce 算子优化笔记
算法