230. 二叉搜索树中第 K 小的元素

中等 含交互动画

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

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

题目描述

给 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 做普通层序遍历正确写法:用中序才能得到升序(层序顺序和大小无关)
  • 错误写法:只按样例推代码正确写法:先写清状态含义和边界条件(样例太少,隐藏用例专打边界)
  • 错误写法:变量名和动画不一致正确写法:代码变量沿用动画里的核心名字(学习时最怕脑内维护两套概念)

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

下一题 →100. 相同的树 ← 返回题库