[剑指 offer]--树--面试题54. 二叉搜索树的第k大节点

1 题目描述

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

2 解题思路

• 方法： 中序遍历 + 提前返回

本文解法基于此性质：二叉搜索树的中序遍历为 递增序列 。

根据以上性质，易得二叉搜索树的 中序遍历倒序 为 递减序列 。
因此，求 “二叉搜索树第 k 大的节点” 可转化为求 “此树的中序遍历倒序的第k 个节点”

• 为求第 k 个节点，需要实现以下 三项工作 ：
  （1）递归遍历时计数，统计当前节点的序号；
  （2）递归到第 k 个节点时，应记录结果 res ；
  （3）记录结果后，后续的遍历即失去意义，应提前终止（即返回）。

• 递归解析：

终止条件： 当节点 root 为空（越过叶节点），则直接返回；
递归右子树： 即 dfs(root.right) ；
三项工作：
（1）提前返回： 若 k=0 ，代表已找到目标节点，无需继续遍历，因此直接返回；
（2）统计序号： 执行 k=k−1 （即从 k 减至 0 ）；
（3）记录结果： 若 k=0 ，代表当前节点为第 k 大的节点，因此记录 res=root.val ；
递归左子树： 即 dfs(root.left) ；

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

作者：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）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

3 解决代码

• 方法： 中序遍历 + 提前返回 《java代码》

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
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);
        //提前返回： 若 k=0 ，代表已找到目标节点，无需继续遍历，因此直接返回
        if(k == 0) return;
        //统计序号： 执行 k=k−1 （即从 k 减至 0 ）；
        k = k - 1;
        //记录结果： 若 k=0 ，代表当前节点为第 k 大的节点，因此记录 res=root.val ；
        if(k == 0){
            res = root.val;
        }
        dfs(root.left);
    }
}

• 方法： 中序遍历 + 提前返回《python代码》

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def kthLargest(self, root: TreeNode, k: int) -> int:
        def dfs(root):
            if not root:
                return 
            dfs(root.right)
            if self.k == 0:
                return 
            self.k = self.k - 1
            if self.k == 0:
                self.res = root.val
            dfs(root.left)
            
        self.k = k
        dfs(root)
        return self.res