《剑指Offer》二叉搜索树的第k个节点——真没你想象中那么难,一招教你秒杀它~

本文介绍了一种在二叉搜索树中查找第K小元素的方法。通过中序遍历可以得到升序序列,进而快速定位目标元素。文章提供了两种实现方式:一种使用额外的顺序表存储遍历结果;另一种采用更高效的递归方式直接找到答案。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

此解为了更容易让大家理解,所以优化版本放在了最后

 思路:对于二叉搜索树而言,中序遍历,就可以得到升序序列,拿到第k小不是秒杀吗?

    public int KthNode (TreeNode root, int k) {
        //对于二叉搜索树而言,中序遍历,就可以由小到大排出val值;
        ArrayList<Integer> array = new ArrayList<>();
        //得到顺序排列
        InitArray(root, array);
        //把不满足条件的排除
        if(root == null || array.size() < k || k == 0){
            return -1;
        }
        return array.get(k - 1);
    }
    //得到顺序排列
    private void InitArray(TreeNode root, ArrayList<Integer> array){
        if(root == null){
            return;
        }
        InitArray(root.left, array);
        array.add(root.val);
        InitArray(root.right, array);
    }

优化:去掉了顺序表,减少了递归层次,但方法不变,依旧秒杀~

    //记录递归次数
    private int count = 0;
    //记录第k小数字
    private int ret = -1;
    public int KthNode (TreeNode root, int k) {
        fundMin(root, k);
        if(ret != -1){
            return ret;
        }else{
            return -1;
        }
    }
    private void fundMin(TreeNode root, int k){
        if(root == null || count > k){
            return;
        }
        fundMin(root.left, k);
        count++;
        //遇到第k小就修改
        if(count == k){
            ret = root.val;
        }
        fundMin(root.right, k);
    }

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

陈亦康

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值