NC81 二叉搜索树的第k个节点（js）

pingting_

于 2022-04-24 22:38:50 发布

阅读量637

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.csdn.net/pingting_/article/details/124393833

算法专栏收录该内容

42 篇文章

订阅专栏

这篇博客介绍了如何在二叉搜索树中找到第k个节点的算法。通过中序遍历，利用二叉排序树的特性，首先找到最小节点，然后计数直到找到第k小的节点。在遍历过程中使用flag和num变量辅助定位，最终返回目标节点的值。这是一个关于数据结构和算法的应用实例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

NC81 二叉搜索树的第k个节点

二叉搜索树又叫二叉排序树，中序遍历结果为升序序列
我们可以通过中序遍历，先找到第k小节点，再计数，计到第k个时，这个节点也就是第k小的节点了，通过ans变量记录这个节点的值并在最后返回
那么如何找到最小的节点呢？根据二叉排序树的性质：左子树的每个节点都小于当前节点，右子树的每个节点都大于当前节点。那么最左边的节点，也就是中序遍历中最先没有左子树的节点最小。
这里设置一个flag，初始化为0，当遍历到了最小节点的时候设置为1，num在 flag==1时才开始计数
当flag==1并且当前num计数到k，也就是第k小的节点，将当前节点的val给ans
最终返回ans

/*
 * function TreeNode(x) {
 *   this.val = x;
 *   this.left = null;
 *   this.right = null;
 * }
 */
/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param proot TreeNode类 
 * @param k int整型 
 * @return int整型
 */
function KthNode( proot ,  k ) {
    if(!proot) return -1
    let flag = 0
    let num = 0
    // ans 记录倒数k小的节点 val
    let ans = -1
    function dfs(node){
        if(node.left) dfs(node.left)
        
        // 当前是最左边的叶子结点  也就是最小的节点
        if(!node.left && flag === 0){
            flag = 1
        }
        if(flag === 1) num ++
        //倒数第k个节点
        if(flag === 1 && num === k ){
            ans = node.val
            num ++
            return
        }
        if(node.right) dfs(node.right)
    }
    dfs(proot)
    return ans
    
}

module.exports = {
    KthNode : KthNode
};