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

题目

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

思路1

• 二叉搜索树的特性就是中序遍历结果为递增序列，而题目要求的是第 k 大节点，所以就应该是要遍历结果为降序，
• 按照先遍历左子树、输出节点、遍历右子树得到的是升序结果，要得到降序，需要按照先遍历右子树、输出节点、再遍历左子树即可
• 什么时候结束递归呢？有两种情况：
  1. root 为 null 的时候，代表遍历到叶子节点了，此时需要返回
  2. 当前节点为第 k 大节点：每遍历到一个节点，将 k 值减1，直到 k 值为 0 ，说明找到第 k 大节点了，此时记录下节点的值，然后开始返回

代码

class Solution {
    int k;
    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;
        }

        dfs(node.right);
        k--;
        if (k == 0) {
            res = node.val;
            return;
        }
        dfs(node.left);
    }
}

复杂度分析

• 时间复杂度：\(O(N)\)
• 空间复杂度：\(O(N)\)