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

题目

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

解题思路

来源：作者：jyd
链接：https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof/solution/mian-shi-ti-54-er-cha-sou-suo-shu-de-di-k-da-jie-d/
来源：力扣（LeetCode）
本文解法基于此性质：二叉搜索树的中序遍历为 递增序列 。
根据以上性质，易得二叉搜索树的 中序遍历倒序 为 递减序列 。
因此，求 “二叉搜索树第 k大的节点” 可转化为求 “此树的中序遍历倒序的第 k 个节点”。

中序遍历 为 “左、根、右” 顺序，递归法代码如下：

// 打印中序遍历
void dfs(TreeNode root) {
    if(root == null) return;
    dfs(root.left); // 左
    System.out.println(root.val); // 根
    dfs(root.right); // 右
}

中序遍历的倒序 为 “右、根、左” 顺序，递归法代码如下：

// 打印中序遍历倒序
void dfs(TreeNode root) {
    if(root == null) return;
    dfs(root.right); // 右
    System.out.println(root.val); // 根
    dfs(root.left); // 左
}

• 为求第 k 个节点，需要实现以下 三项工作 ：

1. 递归遍历时计数，统计当前节点的序号；
2. 递归到第 kk 个节点时，应记录结果 resres ；
3. 记录结果后，后续的遍历即失去意义，应提前终止（即返回）。
   递归解析：
4. 终止条件： 当节点 rootroot 为空（越过叶节点），则直接返回；
5. 递归右子树： 即 dfs(root.right)dfs(root.right) ；
6. 三项工作：
   提前返回： 若 k = 0 ，代表已找到目标节点，无需继续遍历，因此直接返回；
   统计序号： 执行 k = k - 1 （即从 k 减至 0 ）；
   记录结果： 若 k = 0 ，代表当前节点为第 k 大的节点，因此记录 res = root.val ；
7. 递归左子树： 即 dfs(root.left) ；

复杂度分析：

• 时间复杂度 O(N) ： 当树退化为链表时（全部为右子节点），无论 k 的值大小，递归深度都为 N ，占用 O(N) 时间。
  空间复杂度 O(N) ： 当树退化为链表时（全部为右子节点），系统使用 O(N) 大小的栈空间。

class Solution {
    int res, k;
    public int kthLargest(TreeNode root, int k) {
        this.k = k;
        dfs(root);
        return res;
    }
    void dfs(TreeNode root) {
        if(root == null) return;
        dfs(root.right);
        if(k == 0) return;
        if(--k == 0) res = root.val;
        dfs(root.left);
    }
}

注意点：
这里说空间复杂度是O(N)，可能是因为递归的原因，虽然这里只定义了两个变量（res 和 k），但是实际上在查询的时候，对于链表来说，传进去的是根节点，依次往下查询，然后还要回退回来，所以JVM不能立刻回收其所遍历的节点，需要都保留信息，这样才能回退回去，如果是满二叉树，那时间复杂度应该就是O((N+1)/2)。