剑指 Offer 54. 二叉搜索树的第k大节点 给定一棵二叉搜索树,请找出其中第k大的节点
文章图片
【笔记|题解八十四】思路:递归
已知这个是一颗二叉搜索树,其特点是:
1、左子树上所有节点的值均小于它的根节点的值;
2、右子树上所有节点的值均大于它的根节点的值;
3、左、右子树也分别为二叉搜树。
我们可以使用中序遍历来遍历二叉树,其顺序一般是左—>中—>右。如果是左子树在前,右子树在后,遍历结果就是由小到大、如果是右子树在前,左子树在后,遍历结果就是由大到小
因为我们要求的是第k大节点,所以我们只需要先遍历右子树,再遍历左子树即可。
另外,我们也不需要遍历完这个二叉树,我们只需要遍历到第k大的就好,所以我们要定义一个全局变量count为k,每遍历到一个节点我们就减一,直到count等于1,这时遍历到的数字就是最大的。
递归到第 k个节点时,应记录结果 res ;
记录结果后,后续的遍历即失去意义,应提前终止(即返回)。
class Solution {
int count;
int res;
public int kthLargest(TreeNode root, int k) {
count = k;
dfs(root);
return res;
}public void dfs(TreeNode root) {
if (root == null || count == 0) return;
dfs(root.right);
if (count == 1) {
res = root.val;
count--;
return;
}
count--;
dfs(root.left);
}
}
推荐阅读
- 笔记|如何在Windows11安装安卓子系统()
- 笔记|C语言数据结构——二叉树的顺序存储和二叉树的遍历
- 2021年下半年《信息系统项目管理师》真题
- 个人理解|【C语言基础之类型转换】
- 学习分享|【C语言函数基础】
- 个人理解|【C语言实现井字棋及电脑落子优化】
- 总结|python爬虫入门
- Python|蓝桥杯 平面切割 Python
- Python|Python 每日一练 二分查找 搜索旋转排序数组 详解
- 笔记|这是一个关于face_recognition和dlib库的安装(亲测有用,毕竟我代码都写出来了)