剑指offer之面试题54：二叉搜索树的第k大节点

最新推荐文章于 2022-05-20 05:48:01 发布

yudidina

最新推荐文章于 2022-05-20 05:48:01 发布

阅读量155

点赞数

分类专栏：剑指offer 文章标签：指针数据结构二叉树算法面试

本文链接：https://blog.csdn.net/qq_43468570/article/details/105492245

版权

剑指offer 专栏收录该内容

23 篇文章 0 订阅

订阅专栏

二叉搜索树的第k大节点

1、题目

给定一颗二叉搜索树，请找出其中第k大的节点。

输入参数：二叉搜索树的根节点、非负整数k

输出结果：二叉搜索树的某个节点，或者是空指针

2、解题

这道题的关键在于知晓二叉搜索树的中序遍历方法，即通过中序遍历二叉树，便可以得到一个递增排序的遍历序列，这样就能很方便的得到第k大节点的值。

解题的具体步骤为：

鲁棒性检查，而后调用递归函数
递归函数内
- 新建一个用于指向目标节点的指针target
- 先递归遍历左子树，将结果赋值给target
- 判断当前节点是否是第k个节点，若是则将当前节点赋值给target
- 若未找到目标，继续递归遍历右子树
- 最后返回目标

3、代码

BinaryTreeNode* KthNode(BinaryTreeNode* pRoot, unsigned int k){
    //鲁棒性检查
    if(pRoot == nullptr || k == 0)
        return nullptr;
    return KthNodeCore(pRoot, k);
}

BinaryTreeNode* KthNodeCore(BinaryTreeNode* pRoot, unsigned int &k){
    //新建一个用于指向目标节点的指针
    BinaryTreeNode* target = nullptr;
    //先递归遍历左子树
    if(pRoot->m_pLeft != nullptr)
        target = KthNodeCore(pRoot->m_pLeft, k);
    //判断当前节点是否是第k个节点
    if(target == nullptr){
        if(k == 1)
            target = pRoot;
        k--;
    }
    //若未找到目标，继续递归遍历右子树
    if(target == nullptr && pRoot->m_pRight != nullptr)
        target = KthNodeCore(pRoot->m_pRight, k);
        
    //最后返回目标
    return target;
}