leetcode 日经贴，Cpp code -Kth Smallest Element in a BST

原创于 2015-07-02 17:23:04 发布 · 360 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 3 个专栏收录

183 篇文章

订阅专栏

leetcode

173 篇文章

订阅专栏

Cpp

152 篇文章

订阅专栏

本文介绍了一种高效算法来找到给定二叉搜索树中第K小的元素，通过递归地遍历树并记录访问节点数量，从而在O(n)时间内完成查找。

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

Kth Smallest Element in a BST

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    pair<TreeNode *, int> findkth(TreeNode *r, int k) {
        if (!r) {
            return make_pair(static_cast<TreeNode *>(NULL), 0);
        }
        pair<TreeNode *, int> leret = findkth(r->left, k);
        if (!leret.first && k == leret.second + 1) {
            leret.first = r;
        }
        if (leret.first) {
            return leret;
        }
        pair<TreeNode *, int> riret = findkth(r->right, k - leret.second - 1);
        if (riret.first) {
            return riret;
        }
        riret.second += leret.second + 1;
        return riret;
    }
    int kthSmallest(TreeNode* root, int k) {
        pair<TreeNode *, int> ret = findkth(root, k);
        return ret.first->val;
    }
};