二叉搜索树的第K大结点

题目:给定一个二叉搜索树,请找出其中第K大的结点。

  • 如果按照中序遍历的顺序遍历一棵二叉搜索树,则遍历序列的数值是递增排序的。只需要用中序遍历算法遍历一棵二叉树搜索树,很容易找出第K大结点。

const BinaryTreeNode* KthNode(const BinaryTreeNode* pRoot, unsigned int k)
{
    if(pRoot == nullptr || k == 0)
        return nullptr;

    return KthNodeCore(pRoot, k);
}

const BinaryTreeNode* KthNodeCore(const BinaryTreeNode* pRoot, unsigned int& k)
{
    const BinaryTreeNode* target = nullptr;

    if(pRoot->m_pLeft != nullptr)
        target = KthNodeCore(pRoot->m_pLeft, k);

    if(target == nullptr)
    {
        if(k == 1)
            target = pRoot;

        k--;
    }

    if(target == nullptr && pRoot->m_pRight != nullptr)
        target = KthNodeCore(pRoot->m_pRight, k);

    return target;
}

你可能感兴趣的:(树,剑指offer)