*LeetCode-Kth Smallest Element in a BST

一个函数来数这个root下一共多少node 非常简单 recursive

然后就是主函数中判断left有多少node的时候需要数右边 想清楚 +1 -1的事情 因为要算上root本身

public class Solution {
    public int kthSmallest(TreeNode root, int k) {
        if ( countNode ( root.left ) < k - 1)
            return kthSmallest( root.right, k - 1 - countNode( root.left) );
        else if ( countNode ( root.left ) >= k)
            return kthSmallest( root.left, k);
        else
            return root.val;
    }
    public int countNode ( TreeNode node ){
        if ( node == null )
            return 0;
        return 1 + countNode ( node.left ) + countNode ( node.right );
    }
}


你可能感兴趣的:(*LeetCode-Kth Smallest Element in a BST)