剑指offer（62）二叉搜索树的第K个节点

Posted 2021-01-02 Blog of Eric Wu

题目描述

给定一棵二叉搜索树，请找出其中的第k小的结点。例如， （5，3，7，2，4，6，8）    中，按结点数值大小顺序第三小结点的值为4。

 

题目分析

首先，我们可以先画图。画完图后我们要想办法从中找出第K小的节点。

因为这是二叉搜索树，我们可以轻易发现它的中序遍历序列就是从小到大排列，也就是我们可以直接中序遍历，同时计数，就可以得到我们想要的节点了。

不过需要注意的是我们的计数变量k应该在函数外面，不然递归进去后回来时是无法获得已经改变了的k值的。

 

代码

function KthNode(pRoot, k)
{
    if(pRoot == null || k == 0){
        return null;
    }
    //为了能追踪k，应该把KthNodeCore函数定义在这里面，k应该在KthNodeCore函数外面
    function KthNodeCore(pRoot){
        let target = null;
        if(pRoot.left != null){
            target = KthNodeCore(pRoot.left, k)
        }
        if(target == null){
            if(k == 1){
                target = pRoot;
            }
            k--; 
        }
        if(target == null && pRoot.right != null){
            target = KthNodeCore(pRoot.right, k);
        }
        return target;
    }
    return KthNodeCore(pRoot);
}