题目

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

示例 1：

输入：root = [3,1,4,null,2], k = 1
输出：1

解

class Solution {
    int count = 0;

    public int kthSmallest(TreeNode root, int k) {
        int[] result = { -1 };
        dfs(root, k, result);
        return result[0];
    }

    public void dfs(TreeNode root, int k, int[] result) {
        if (root == null) {
            return;
        }

        dfs(root.left, k, result);
        count++;
        if (count == k) {
            result[0] = root.val;
        }
        dfs(root.right, k, result);
    }
}