剑指 offer 54. 二叉搜索树的第k大节点

剑指 Offer 54. 二叉搜索树的第k大节点

给定一棵二叉搜索树，请找出其中第k大的节点。

示例 1:

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 4

示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 4

限制：

• 1 ≤ k ≤ 二叉搜索树元素个数

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

Solution 1

class Solution {
    private int count = 0;

    public int kthLargest(TreeNode root, int k) {
        count = k;
        TreeNode node = inorderSearch(root);
        // if (node == null)
        // throw new NullPointerException();
        return node.val;
    }

    private TreeNode inorderSearch(TreeNode root) {
        if (root == null)
            return null;
        TreeNode node = null;
        node = inorderSearch(root.right);
        // early stop
        if (node != null)
            return node;
        // curent node is kthLargest node
        if (--count == 0)
            return root;
        node = inorderSearch(root.left);
        return node;
    }
}