二叉树搜索第k个节点（41）

Xu_lib

于 2020-03-21 19:53:07 发布

阅读量280

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指Offer

本文链接：https://blog.youkuaiyun.com/Jeffxu_lib/article/details/105015920

剑指Offer 专栏收录该内容

42 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中寻找第K小结点的算法，利用二叉搜索树的特性，通过中序遍历的方式，可以有效地找到目标结点。文章详细解析了算法的实现思路，并提供了具体的代码实现。

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

题目

【给定一棵二叉搜索树，请找出其中的第k小的结点。例如，（5，3，7，2，4，6，8）中，按结点数值大小顺序第三小结点的值为4】

1、分析
二叉搜索树的左子树都小于根节点，右子树都大于根节点。
在这里插入图片描述
且其中序遍历刚好是由大到小的顺序，所以用中序遍历来对二叉搜索树进行遍历，同时统计第k大的节点
2、代码

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot==nullptr || k<1)
            return nullptr;
        return FindKthNode(pRoot,k);
    }
    
    TreeNode* FindKthNode(TreeNode* pRoot,int &k)
    {
        TreeNode *target=nullptr;
        if(pRoot->left!=nullptr)
            target=FindKthNode(pRoot->left,k);
        if(target==nullptr)
        {
            if(k==1)
                return pRoot;
            k--;
        }
        if( target==nullptr && pRoot->right!=nullptr)
        {
            target=FindKthNode(pRoot->right,k);
        }
        
        return target;
    }
    

    
};