二叉搜索树的第K个结点

外界定义一个标识变量来记录中序访问到第几个节点,当访问到第k个节点是返回结果。注意第五行和第七行的i<k条件加上是为了在找到第K个数就终止递归,防止遍历整个二叉树。

int i = 0;
TreeNode resNode = null;
TreeNode KthNode(TreeNode pRoot, int k) {
    if (pRoot == null) return null;
    if (i < k && pRoot.left != null) KthNode(pRoot.left, k);
    if (++i == k) resNode = pRoot;
    if (i < k && pRoot.right != null) KthNode(pRoot.right, k);
    return resNode;
}
全部评论

相关推荐

昨天 22:21
门头沟学院 Java
点赞 评论 收藏
分享
12-13 14:51
已编辑
井冈山大学 算法工程师
龙虾x:算法比你强的没有你美,比你美的…..算了已经没有比你美的了
工作两年想退休了
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务