二叉搜索树的第k个结点

最新推荐文章于 2022-04-07 23:30:53 发布

原创最新推荐文章于 2022-04-07 23:30:53 发布 · 330 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #遍历

算法题专栏收录该内容

40 篇文章

订阅专栏

题目：

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

分析：
中序遍历二叉树。

class Solution
{
public:
    TreeNode* KthNode( TreeNode* pRoot, int k )
    {
        if ( pRoot == NULL || k == 0 )
            return NULL;
        return KthNodeCore( pRoot, k );
    }
    TreeNode* KthNodeCore( TreeNode* pRoot, int& k )
    {
        TreeNode* target = NULL;
        if ( pRoot->left != NULL )
            target = KthNodeCore( pRoot->left, k );
        if ( target == NULL )
        {
            if ( k == 1 )
                target = pRoot;
            k--;
        }
        if ( target == NULL && pRoot->right != NULL )
            target = KthNodeCore( pRoot->right, k);

        return target;
    }
};