题目描述
对 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 映射(重复扫描会变慢)
- 错误写法:只按样例推代码 → 正确写法:先写清状态含义和边界条件(样例太少,隐藏用例专打边界)
- 错误写法:变量名和动画不一致 → 正确写法:代码变量沿用动画里的核心名字(学习时最怕脑内维护两套概念)
以上文字与上方动画完全同一思路:先看动画建立直觉 → 读文字巩固 → 关掉页面自己默写一遍代码,能写出来才算真的掌握。