【Java实现】剑指Offer54——二叉搜索树的第K大节点(利用BST的性质遍历)思路分享

【Java实现】剑指Offer54——二叉搜索树的第K大节点(利用BST的性质遍历):思路分享

图片来自LeetCode
关于二叉搜索树的一些学习笔记在这里哦:

接下来几期内容会更新《剑指Offer》中所有“树”标签的习题及思路,传送门在此:


解题思路:
从题目看到二叉搜索树,就要联想到BST的性质:

中序遍历是递增的

我们都知道二叉树的中序遍历:

先遍历左子树,然后根节点,最后右子树

既然BST这样遍历能得到递增序列,那如何得到递减序列呢?

先遍历右子树、然后根节点、最后左子树,即可得到递减序列

到这里,大体的思路已经确定,就是采用递归的方式遍历BST,而要获得第k大的数,只需要加一个计数器即可。

思路如下:

  1. 每遍历到一个数,都要判断当前计数器的状态
  2. 假设计数器初始值为k,当计数器归零时,即得到结果
  3. 此时递归即可终止,返回值即可

代码如下:

class Solution {
    //需要定义全局变量,计数器和结果
    private int count,res;

    public int kthLargest(TreeNode root, int k) {
        //让计数器递减
        this.count=k;
        dfs(root);
        return res;
    }
    public void dfs(TreeNode root) {
        //递归终止条件
        if(root==null) return;
        //先遍历右子树
        dfs(root.right);

        //回来遍历当前节点,通过count判断
        if(count==0) return;
        if(--count==0) res=root.val;

        //最后递归左子树
        dfs(root.left);
    }
}

需要注意的点在于,计数器count--操作后,刚好等于0,此时是要返回的值,需要掌握好这个时机。

复杂度分析:

  • 时间:O(N),需要遍历树
  • 空间:O(N),递归占用系统栈内存
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值