【剑指offer】二叉搜索树的第k个结点（二叉树）

最新推荐文章于 2022-04-27 18:36:50 发布

zjwreal

最新推荐文章于 2022-04-27 18:36:50 发布

阅读量207

点赞数

分类专栏：剑指offer

本文链接：https://blog.csdn.net/zjwreal/article/details/89289371

版权

剑指offer 专栏收录该内容

80 篇文章 4 订阅

订阅专栏

题目描述

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

思路

对于二叉搜索树，使用中序遍历就能得到排序好的序列；每次访问一个节点，计数-1；直到计数为0时返回。

代码

/*
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 find(pRoot, k);
    }

    TreeNode *find(TreeNode* pRoot, int &k){
        if(pRoot == nullptr)
            return nullptr;

        TreeNode *left = find(pRoot->left,k);
        if(left!= nullptr){
            if(k==0)    // 是0直接返回
                return left;
            --k;
        }
        
        if(--k == 0)    // 注意这个地方要先减1再判断
            return pRoot;

        TreeNode *right = find(pRoot->right,k);
        if(right!= nullptr){
            if(k == 0)
                return right;
            --k;
        }
        return nullptr;
    }
};