【LeetCode】剑指 Offer 54. 二叉搜索树的第k大节点
文章目录
- 【LeetCode】剑指 Offer 54. 二叉搜索树的第k大节点
- 一、中序遍历（提前返回）
- 总结
一、中序遍历（提前返回）
二叉搜索树的中序遍历为递增序列
根据这个性质，易得二叉搜索树的中序遍历倒序为递减序列，因此，求 “二叉搜索树第 k 大的结点” 可转化为求 “此树的中序遍历倒序的第 k 个结点”
为了求第 k 个结点，需要实现以下三项工作：
- 递归遍历时计数，统计当前结点的序号
- 递归到第 k 个结点时，应记录结果 res
- 记录结果后，后续的遍历即失去意义，应提前终止
递归解析：
- 终止条件：当结点 root 为空（越过叶结点），则直接返回
- 递归右子树：即 dfs(root.right)
- 三项工作：
- 提前返回：若 k = 0，代表已找到目标结点，无需继续遍历，因此直接返回
- 统计序号：执行 k = k - 1
- 记录结果：若 k = 0，代表当前结点为第 k 大的结点，因此记录 res = root.val
- 递归左子树：即 dfs(root.left)
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);
}
}
总结
K 神 yyds
