【剑指offer54.二叉搜索树的第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 ≤ 二叉搜索树元素个数 
// Related Topics 树 
// 👍 131 👎 0

class Solution {
    private int kthLargestNum = 0;
    private int count= 0;
    public int kthLargest(TreeNode root, int k) {
        if (root == null) {
            return 0;
        }

        count = k;
        // 遍历，从右向左，遍历到第k个子节点时，即为第K大的节点
        dfsTree(root, k);
        return kthLargestNum;
    }

    private void dfsTree(TreeNode root, int k) {
        if (root == null) {
            return;
        }

        // 先右，再到根，再到左
        dfsTree(root.right, count);
        count--;
        if (count == 0) {
            kthLargestNum = root.val;
            return;
        }

        dfsTree(root.left, count);
    }
}