题目链接

示例

给定一棵二叉搜索树,请找出其中第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

解题思路

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

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

为求第 k 个节点,需要实现以下 三项工作 :

  1. 递归遍历时计数,统计当前节点的序号;
  2. 递归到第 k 个节点时,应记录结果 res ;
  3. 记录结果后,后续的遍历即失去意义,应提前终止(即返回)。

递归解析:

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

代码

  1. class Solution {
  2. int res, k;
  3. public int kthLargest(TreeNode root, int k) {
  4. this.k = k;
  5. dfs(root);
  6. return res;
  7. }
  8. void dfs(TreeNode root) {
  9. if(root == null) return;
  10. dfs(root.right);
  11. if(k == 0) return;
  12. if(--k == 0) res = root.val;
  13. dfs(root.left);
  14. }
  15. }