496. 下一个更大元素 I

简单 含交互动画

学完这道你应该能: 说清它为什么用「单调栈」,而不是死记步骤; 合上页面,自己默写出核心代码; 用 30 秒把思路讲清楚。

AI 私教 · 就着本题动画讲
卡住了不用硬扛,按 8 步把这题真正走通。
从读题、试解、找法到手写代码,边问边打勾,适合第一次系统学算法的同学。

题目描述

对 nums1 中每个数,找它在 nums2 中右侧第一个更大元素。

nums1 = [4,1,2]
nums2 = [1,3,4,2]
输出 = [-1,3,-1]

思路解析

关键不是背模板,而是看懂为什么这题适合单调栈

1. 扫描 nums2 的 1,入栈等待更大值:扫描 nums2 的 1,入栈等待更大值

2. 遇到 3,弹出 1,记录 1→3:遇到 3,弹出 1,记录 1→3

3. 3 入栈继续等:3 入栈继续等

4. 遇到 4,弹出 3,记录 3→4:遇到 4,弹出 3,记录 3→4

5. 4 入栈,后面没有更大:4 入栈,后面没有更大

6. 遇到 2,无法弹 4,2 入栈:遇到 2,无法弹 4,2 入栈

7. 栈里剩下的都没有更大,记 -1:栈里剩下的都没有更大,记 -1

8. 按 nums1 查询映射表:按 nums1 查询映射表

把这句话记住,下次遇到同类题,就能更快选出方向。

用一个小问题检查自己是不是真的懂了。

参考代码(Python)

Python
1class Solution:
2    def nextGreaterElement(self, nums1, nums2):
3        mp = {}
4        st = []
5        for x in nums2:
6            while st and st[-1] < x:
7                mp[st.pop()] = x
8            st.append(x)
9        return [mp.get(x, -1) for x in nums1]

复杂度分析

  • 时间复杂度:O(m+n) —— 每个核心状态按算法要求处理固定次数
  • 空间复杂度:O(n) —— 只保存必要的辅助结构或递归栈

套路模板

模板不是死背,而是提醒你写代码前先把状态、转移和边界排好。

Python
1# 单调栈 通用检查表
2# 1. 定义状态/指针/容器
3# 2. 每轮只做一个清晰动作
4# 3. 更新答案并处理边界

易错点

  • 错误写法:对 nums1 每个数都去 nums2 右边扫正确写法:先预处理 nums2 的 next greater 映射(重复扫描会变慢)
  • 错误写法:只按样例推代码正确写法:先写清状态含义和边界条件(样例太少,隐藏用例专打边界)
  • 错误写法:变量名和动画不一致正确写法:代码变量沿用动画里的核心名字(学习时最怕脑内维护两套概念)

以上文字与上方动画完全同一思路:先看动画建立直觉 → 读文字巩固 → 关掉页面自己默写一遍代码,能写出来才算真的掌握

下一题 →402. 移掉 K 位数字 ← 返回题库