二叉搜索树的第k个结点

最新推荐文章于 2020-12-06 21:23:22 发布

俺也一样

最新推荐文章于 2020-12-06 21:23:22 发布

阅读量114

点赞数

分类专栏：剑指offer

本文链接：https://blog.csdn.net/weixin_41791402/article/details/100084620

版权

剑指offer 专栏收录该内容

38 篇文章 0 订阅

订阅专栏

题目描述

给定一棵二叉搜索树，请找出其中的第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:
    vector<int> nums;
    TreeNode* ret;
    TreeNode* KthNode(TreeNode* root, int k)
    {
        ret = NULL;
        if(root == NULL || k == 0)
            return NULL;
        bianli(root);
        sort(nums.begin(), nums.end());
        int val = nums[k-1];
        cal(root, val);
        return ret;
    }
    
    void bianli(TreeNode* root)
    {
        if(root != NULL)
        {
            nums.push_back(root->val);
            bianli(root->left);
            bianli(root->right);
        }
    }
    void cal(TreeNode* root,int& val)
    {
        if(root != NULL)
        {
            if(ret == NULL)
            {
                if(root->val == val)
                {
                    ret = root;
                    return;
                }
            }
            else
                return;
            
            
            cal(root->left,val);
            cal(root->right,val);
        }
    }
    
};