【java】剑指offer54_二叉搜索树的第K大节点

122 篇文章 0 订阅

题目描述:

给定一棵二叉搜索树,请找出其中第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
限制:1 ≤ k ≤ 二叉搜索树元素个数

参考解题思路:利用二叉搜索树的特性(左节点<根节点<右节点),又因为求第K大的树,所以反向中序遍历二叉树,先右节点再根节点再左节点即可,再利用一个全局变量记录每次求得的节点值与目标K是否相等,相等则利用全局变量res赋值当前节点的val即可。定义一个方法helper反向中序遍历二叉搜索树

    int num = 0;
    int res = 0;
    public int kthLargest(TreeNode root, int k) {
        num = k;
        helper(root);
        return res;
    }

    public void helper(TreeNode root) {
        // 遍历右节点
        if (root.right != null && num >0) {
            helper(root.right);
        }
        // 判断已经遍历节点是否满足目标K值
        num--;
        if (num == 0) {
            // 目标节点赋值返回
            res = root.val;
            return;
        }
        // 未满足目标K值继续遍历二叉搜索树的左节点
        if (root.left != null && num > 0) {
            helper(root.left);
        }
    }

复杂度分析:
时间复杂度 O(N): 当树退化为链表时(全部为右子节点),无论 k 的值大小,递归深度都为 N ,占用 O(N) 时间。
空间复杂度 O(N): 当树退化为链表时(全部为右子节点),系统使用 O(N)大小的栈空间。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值