题解八十四

最新推荐文章于 2020-07-20 11:02:26 发布

Yuzuru Hanyu's pooh

最新推荐文章于 2020-07-20 11:02:26 发布

阅读量165

点赞数

分类专栏：笔记总结文章标签：二叉树数据结构 dfs 算法 leetcode

本文链接：https://blog.csdn.net/qq_45138241/article/details/107355749

版权

笔记同时被 2 个专栏收录

101 篇文章 0 订阅

订阅专栏

总结

100 篇文章 0 订阅

订阅专栏

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