算法 - 剑指Offer 二叉搜索树的第k大节点

题目

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

解题思路

这题比较简单，首先题目中说需要倒数第几大的数字， 第一反应就是这个坑定是一个倒序数组，然后去找， 然后我们可以知道的是一般树结构遍历中序遍历二叉搜索树的时候都是左-》根-》右顺序是真好正序，但是如果我们遍历的时候将右-》根-》左则正好满足我们的倒序，然后我们只需要按照K去判断即可， 这里还有一个坑的点，这里的k不能当做变量传进递归函数中， 因为我们要的K是一直减的， 但是递归里面的K是返回上一值得， 可能会存在K=1有两个root值， 这里不方便计算， 所以我们将K附到全局变量上即可， 下面的代码即可。

Java代码实现

public class KthLargest {
    public class TreeNode {
       int val;
       TreeNode left;
       TreeNode right;
       TreeNode(int x) { val = x; }
   }
    int res = -1;
    int k = -1;
    public int kthLargest(TreeNode root, int tar) {
        k = tar;
        dfs(root);
        return res;
    }
    private void dfs(TreeNode root) {
        if(root == null || k == 0){
            return ;
        }
        dfs(root.right);
        k = k - 1;
        if(k == 0){
            res = root.val;
        }
        dfs(root.left);
    }
}