leetcode刷题. 230. 二叉搜索树中第K小的元素

BST树中序遍历找第K小元素

最新推荐文章于 2022-11-13 18:57:18 发布

原创最新推荐文章于 2022-11-13 18:57:18 发布 · 142 阅读

0 ·

CC 4.0 BY-SA版权

【算法】同时被 2 个专栏收录

114 篇文章

订阅专栏

【leetcode刷题】

95 篇文章

订阅专栏

本文介绍了一种在二叉搜索树(BST)中寻找第k小元素的方法，通过中序遍历的方式，利用BST的性质，找到目标节点。此方法适用于需要在有序序列中查找特定位置元素的场景。

因为此处是BST树，所以中序遍历是升序的有序遍历，所遍历的第k个节点即是目标节点。

此是对二叉树遍历的一点复习吧。

    void middleTraverse(TreeNode *root, int k, int &count, int &val) {

        if(count > k || nullptr == root)
            return;
        
        middleTraverse(root->left, k, count, val);

        if(++count == k) {

            val = root->val;
            return;
        }
        middleTraverse(root->right, k, count, val);
    }

    int kthSmallest(TreeNode* root, int k) {

        int count = 0, val = 0;
        middleTraverse(root, k, count, val);
        return val;
    }