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

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

解题思路

对于二叉搜索树来说，中序遍历就是从小到大遍历节点值，所以遍历到第 k 个节点即是答案

参考代码

class Solution {
    private int res;
    private int k;

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

    private void dfs(TreeNode node) {
        if(node == null || k < 0) {
            return;
        }
        dfs(node.left);
        if(k > 0) {
            k --;
            res = node.val;
        }
        dfs(node.right);
    }
}