【面试经典 150 | 二叉搜索树】二叉搜索树中第K小的元素

写在前面

本专栏专注于分析与讲解【面试经典150】算法，两到三天更新一篇文章，欢迎催更……

专栏内容以分析题目为主，并附带一些对于本题涉及到的数据结构等内容进行回顾与总结，文章结构大致如下，部分内容会有增删：

• Tag：介绍本题牵涉到的知识点、数据结构；
• 题目来源：贴上题目的链接，方便大家查找题目并完成练习；
• 题目解读：复述题目（确保自己真的理解题目意思），并强调一些题目重点信息；
• 解题思路：介绍一些解题思路，每种解题思路包括思路讲解、实现代码以及复杂度分析；
• 知识回忆：针对今天介绍的题目中的重点内容、数据结构进行回顾总结。

---

Tag

【二叉搜索树】【中序遍历】

---

题目来源

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

---

解题思路

方法一：中序遍历

利用二叉搜索树的性质可知，中序遍历后会得到递增的子序列。我们可以在遍历时将值序列存放在数组中，然后通过索引 k-1 找出第 k 小的数。这种方法比较基础，不做详细介绍了。

思路

我们还可以利用迭代对二叉搜索树进行中序遍历，在遍历的过程中找到第 k 小的数。只需要将迭代实现二叉树中序遍历中的记录当前遍历的节点值改成以下代码即可：

--k;
if(k == 0){
    break;
}

完整实现代码如下所示。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int kthSmallest(TreeNode* root, int k) {
        stack<TreeNode*> stk;

        while(root || !stk.empty()){
            while(root){
                stk.push(root);
                root = root->left;
            }
            root = stk.top(); stk.pop();
            --k;
            if(k == 0){
                break;
            }
            root = root->right;
        }
        return root->val;
    }
};

复杂度分析

时间复杂度：$O(H+k)$，$H$ 是二叉搜索树的高度。我们需要 $O(H)$ 的时间到达叶节点。

空间复杂度：$O(N)$，最坏的情况下二叉搜索树退化成一条链，节点数记为 $N$，空间复杂度取得最大值 $O(N)$。

---

写在最后

如果您发现文章有任何错误或者对文章有任何疑问，欢迎私信博主或者在评论区指出 💬💬💬。

如果大家有更优的时间、空间复杂度的方法，欢迎评论区交流。

最后，感谢您的阅读，如果有所收获的话可以给我点一个 👍 哦。