面试算法-123-二叉搜索树中第K小的元素

题目

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

示例 1：

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

题目

class Solution {
    public int kthSmallest(TreeNode root, int k) {
        dfs(root,k);
        return node.val;
    }

    int count = 0;
    TreeNode node = null;
    public void dfs(TreeNode root, int k){
        if(root == null){
            return;
        }
        
        dfs(root.left,k);
        count++;
        if(count == k){
            node = root;
            return;
        }
        dfs(root.right,k);
    }
}