求二叉排序树的第k个节点

记录下自己愚蠢的一天
首先题目一看都能想到要用中序遍历打方法

int index=0;
TreeNode* KthNode(TreeNode* root, int k)
{
    if(root != NULL)
    { //中序遍历寻找第k个
     KthNode(root->left,k);
        index++;
        if(index == k)
      return root;
        KthNode(root->right,k);
    }
    return NULL;
}

这是我一开始以为没问题的第一段代码,中序遍历嘛,找到第k个就返回,可是我陷入了一个误区,当index==k时,返回的指针只是其中一个递归函数的返回值,并不是说我们最初的这个函数返回了这个指针!
因为如果我们要找的那个节点是在其递归深度中的某个时候找到的。
要这么理解这个思路,对于根节点root,在其左子树中如果找到第k大的节点,则输出该节点,否则判断它自己是不是第k大的节点,然后递归右子树。
代码如下:

int index=0;
    TreeNode* KthNode(TreeNode* root, int k)
    {
        if(root != NULL)
        { //中序遍历寻找第k个
            TreeNode* node = KthNode(root->left,k);
            if(node != NULL)//找到该节点了
                return node;
            index++;
            if(index == k)
                return root;
            node = KthNode(root->right,k);
            if(node != NULL)//找到该节点
                return node;
        }
        //没有找到就返回null
        return NULL;
    }
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值