牛客剑指Offer面试题54:二叉搜素树的第ｋ大节点

题目描述

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

思路

根据二叉搜索树的特点，其中序遍历结果即按从小到大的顺序排列各节点．
按左根右的顺序遍历节点，每遍历一个节点（调用一次递归函数），距离第ｋ小节点接近了一步，再传入递归函数的ｋ就减１．直至ｋ＝＝１时，当前节点即为所求

代码

/*
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 == 0)
            return nullptr;
        return KthNodeCore(pRoot, k);
    }
private:
    TreeNode* KthNodeCore(TreeNode* pRoot, int& k)
    {
        //定义目标节点
        TreeNode* targetNode = nullptr;
        //如果当前节点存在左子节点，递归遍历左子树
        if(pRoot->left != nullptr)
            targetNode = KthNodeCore(pRoot->left, k);
        //如果此时还未找到目标节点
        if(targetNode == nullptr)
        {
            //若k=1,当前节点即为所求
            if(k == 1)
                targetNode = pRoot;
            //未找到目标节点，离目标节点近了一步
            k--;
        }
        //当前节点左子节点和当前节点均为找到目标节点，递归寻找右子节点
        if(targetNode == nullptr && pRoot->right != nullptr)
            targetNode = KthNodeCore(pRoot->right, k);
        return targetNode;
    }
    
    
};