leetcode 230 Kth Smallest Element in a BST C++

最新推荐文章于 2024-03-04 15:18:28 发布

wcxdell

最新推荐文章于 2024-03-04 15:18:28 发布

阅读量644

点赞数

CC 4.0 BY-SA版权

分类专栏：算法 C++ 文章标签： c++ LeetCode 算法

本文链接：https://blog.youkuaiyun.com/a2331046/article/details/70149124

C++ 同时被 2 个专栏收录

133 篇文章

订阅专栏

算法

46 篇文章

订阅专栏

本文介绍了一种寻找二叉搜索树中第K小元素的高效算法。首先通过中序遍历将节点值存储到数组中，然后直接返回第K个元素。为了提高效率，还提供了一个改进版的方法，在递归过程中直接找到目标节点，避免了完整的中序遍历。

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

先用简单的方法，中序遍历放到一个数组里面，取k-1即可。

    vector<int> vec;
    void inorder(TreeNode *root) {
        if (!root) return;
        inorder(root->left);
        vec.push_back(root->val);
        inorder(root->right);
    }
    int kthSmallest(TreeNode* root, int k) {
        inorder(root);
        return vec[k-1];
    }

但是效率比较低。

后来用了这种，直接在递归的过程中找，因为每次遍历都相当于拿到了一个节点，并且是递增的。

    int val = 0;
    int sum = 0;
    int numK;
    void inorder(TreeNode *root) {
        if(!root) return;
        inorder(root->left);
        if(++sum == numK) {
          val = root->val;
          return;
        } 
        inorder(root->right);
    }
    int kthSmallest(TreeNode* root, int k) {
        numK = k;
        inorder(root);
        return val;
    }