题目描述
给 BST 和 k,返回第 k 小的节点值。
root = [3,1,4,null,2], k=1
输出 = 1
思路解析
关键不是背模板,而是看懂为什么这题适合BST · 中序遍历。
1. 从根一路压左节点:从根一路压左节点
2. 最左节点 1 是第 1 小:最左节点 1 是第 1 小
3. 如果 k=1,访问 1 就返回:如果 k=1,访问 1 就返回
4. 弹栈访问当前节点:弹栈访问当前节点
5. 访问后转向右子树:访问后转向右子树
6. 右子树也按同样规则压左:右子树也按同样规则压左
7. 计数访问次数:计数访问次数
8. 第 k 次访问就是答案:第 k 次访问就是答案
把这句话记住,下次遇到同类题,就能更快选出方向。
用一个小问题检查自己是不是真的懂了。
参考代码(Python)
Python
1class Solution:
2 def kthSmallest(self, root, k):
3 stack = []
4 cur = root
5 while cur or stack:
6 while cur:
7 stack.append(cur)
8 cur = cur.left
9 cur = stack.pop()
10 k -= 1
11 if k == 0:
12 return cur.val
13 cur = cur.right复杂度分析
- 时间复杂度:O(h+k) —— 每个核心状态按算法要求处理固定次数
- 空间复杂度:O(h) —— 只保存必要的辅助结构或递归栈
套路模板
模板不是死背,而是提醒你写代码前先把状态、转移和边界排好。
Python
1# BST · 中序遍历 通用检查表
2# 1. 定义状态/指针/容器
3# 2. 每轮只做一个清晰动作
4# 3. 更新答案并处理边界易错点
- 错误写法:对 BST 做普通层序遍历 → 正确写法:用中序才能得到升序(层序顺序和大小无关)
- 错误写法:只按样例推代码 → 正确写法:先写清状态含义和边界条件(样例太少,隐藏用例专打边界)
- 错误写法:变量名和动画不一致 → 正确写法:代码变量沿用动画里的核心名字(学习时最怕脑内维护两套概念)
以上文字与上方动画完全同一思路:先看动画建立直觉 → 读文字巩固 → 关掉页面自己默写一遍代码,能写出来才算真的掌握。