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

最新推荐文章于 2024-07-09 13:30:51 发布

weixin_45440200

最新推荐文章于 2024-07-09 13:30:51 发布

阅读量83

点赞数

分类专栏：算法文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_45440200/article/details/120257039

版权

算法专栏收录该内容

39 篇文章 0 订阅

订阅专栏

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

难度中等445

给定一个二叉搜索树的根节点 root ，和一个整数 k ，请你设计一个算法查找其中第 k 个最小元素（从 1 开始计数）。

示例 1：

输入：root = [3,1,4,null,2], k = 1
输出：1

示例 2：

输入：root = [5,3,6,2,4,null,null,1], k = 3
输出：3

提示：

树中的节点数为 n 。
1 <= k <= n <= 104
0 <= Node.val <= 104

**进阶：**如果二叉搜索树经常被修改（插入/删除操作）并且你需要频繁地查找第 k 小的值，你将如何优化算法？

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} k
 * @return {number}
 */
var kthSmallest = function(root, k) {
    let index = 1;
    let val = 0;
    let find = function(root){
        if(!root) return ;
        if(root.left) find(root.left);
        if(index==k) {
            // console.log(index, k, root.val);
            val =  root.val;
        }else if(index>k) return ;
        index++;
        if(root.right) find(root.right);
    }
    // let 
    find(root)
    return val;
};