【DFS】LeetCode 230. 二叉搜索树中第K小的元素

题目链接

230. 二叉搜索树中第K小的元素

思路

利用 BST 中序遍历是有序的特点，同时每遍历完一个结点将 k--，当 k = 0 的时候，当前结点是第 k 个结点

代码

class Solution {
    private int k;
    private int result;

    public int kthSmallest(TreeNode root, int k) {
        this.k = k;
        dfs(root);

        return this.result;
    }

    private void dfs(TreeNode root) {
        if(root == null || k <= 0){
            return;
        }

        dfs(root.left);
        k--;

        if(k == 0){
            result = root.val;
        }
        dfs(root.right);
    }
}