Java算法_ BST 中第 k 个最小元素 (LeetCode_Hot100)

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

获得更多？算法思路:代码文档，算法解析的私得。

运行效果

完整代码

/**
 * 2 * @Author: LJJ
 * 3 * @Date: 2023/8/21 13:31
 * 4
 */
public class KthSmallestElement {
    static class TreeNode{
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int val){
            this.val = val;
        }
    }

    private int count = 0; // 计数器，用于记录已经访问的节点个数
    private int result = 0; // 存储第 k 个最小元素的值

    public int kthSmallest(TreeNode root, int k){
        inorderTraversal(root,k); //开始中序遍历查找第k个最小元素
        return result;
    }

    private void inorderTraversal(TreeNode node, int k){
        if (node == null || count >= k){
            return; //诋毁终止条件，节点为空或计数器达到k
        }
        inorderTraversal(node.left, k); // 递归遍历左子树

        count++; //计数器加一
        if (count == k){
            result = node.val;      // 如果计数器等于 k，说明找到第 k 个最小元素
        }
        inorderTraversal(node.right, k); // 递归遍历右子树
    }

    public static void main(String[] args) {
        // 构建二叉搜索树
        TreeNode root = new TreeNode(5);
        root.left = new TreeNode(2);
        root.left.right = new TreeNode(4);
        root.right = new TreeNode(9);
        root.right.left = new TreeNode(6);
        root.right.right= new TreeNode(10);
        KthSmallestElement solution = new KthSmallestElement();
        int k = 3; // 要找第 k 个最小元素
        int result = solution.kthSmallest(root, k);
        System.out.println("第 " + k + " 个最小元素是: " + result); // 输出结果
    }
}