笔记|题解八十四

剑指 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); } }

    推荐阅读