剑指offer -- 二叉搜索树的第 k 个结点

最新推荐文章于 2020-08-08 11:48:34 发布

原创最新推荐文章于 2020-08-08 11:48:34 发布 · 161 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer

算法专栏收录该内容

212 篇文章

订阅专栏

本文探讨了在二叉搜索树中查找第K小节点的有效算法。通过中序遍历，可以确保节点按升序访问，从而在遍历过程中找到所需的第K小节点。文章提供了一个具体的解决方案，并附带了实现该算法的代码示例。

题目描述

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

解题思路

中序遍历二叉搜索树，结果为树中结点从小到大的顺序排列。
一边按照中序遍历二叉搜索树一边记录个数。

AC代码

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    int order = 0;
    TreeNode kth = null;
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null) return kth;
        kthNodeCore(pRoot, k);
        return kth;
    }
    
    private void kthNodeCore(TreeNode pRoot, int k){
        if(pRoot == null) return;
        kthNodeCore(pRoot.left, k);
        order++;
        if(order == k) kth = pRoot;
        kthNodeCore(pRoot.right, k);
    }
}