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

二叉搜索树的第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
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof

解题思路

1、此树为一颗二叉搜索树，若左子树不为空，根节点左侧所有节点的值小于根节点的值；若右子树不为空，右侧所有节点的值大于根节点的值。左右子树也都为二叉搜索树。
2、递归顺序：右子树，根节点，左子树。值得注意的是：递归里有一个判断count == k,如果不做此判断，会遍历所有节点，增加时间复杂度。

代码1

class Solution {
    int k;
    int count;
    int res;
    public int kthLargest(TreeNode root, int k) {
        this.k = k;
        dfs(root);
        return res;

    }
   public void  dfs(TreeNode node){
       if(node == null) return;
       if(node.right != null){
           dfs(node.right);
       }
       if(count == k) return ;//如果不做此判断，无法提前返回，会遍历所有节点，
       if(++count == k){
           res=node.val;
           return; //此时已经满足条件返回，假设返回到 dfs(node.right);如果忽略count == k判断，++count != k,则一直不会返回，会遍历完所有节点，增加时间复杂度
       } 
       if(node.left != null){
           dfs(node.left);
       }
   }
}