算法——二叉搜索树转换成双向链表c++实现

最新推荐文章于 2022-04-28 18:37:28 发布

mmtdggd

最新推荐文章于 2022-04-28 18:37:28 发布

阅读量317

点赞数

分类专栏：算法 C++

本文链接：https://blog.csdn.net/qq_36358357/article/details/105393513

版权

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

15 篇文章 0 订阅

订阅专栏

算法

1 篇文章 0 订阅

订阅专栏

题目：将二叉搜索树换成双向链表

思路：利用树的中序遍历（左根右），再利用两个“快慢”指针，pLast比pCur慢一步，通过这两个指针建立双向链表

BinaryTreeNode* Convert(BinaryTreeNode* pRootOfTree)
{
    BinaryTreeNode *pLastNodeInList = nullptr;
    ConvertNode(pRootOfTree, &pLastNodeInList);

    // pLastNodeInList指向双向链表的尾结点，
    // 我们需要返回头结点
    BinaryTreeNode *pHeadOfList = pLastNodeInList;
    while(pHeadOfList != nullptr && pHeadOfList->m_pLeft != nullptr)
        pHeadOfList = pHeadOfList->m_pLeft;

    return pHeadOfList;
}

void ConvertNode(BinaryTreeNode* pNode, BinaryTreeNode** pLastNodeInList)
{
    if(pNode == nullptr)
        return;

    BinaryTreeNode *pCurrent = pNode;

    if (pCurrent->m_pLeft != nullptr)
        ConvertNode(pCurrent->m_pLeft, pLastNodeInList);

    pCurrent->m_pLeft = *pLastNodeInList; 
    if(*pLastNodeInList != nullptr)
        (*pLastNodeInList)->m_pRight = pCurrent;

    *pLastNodeInList = pCurrent;

    if (pCurrent->m_pRight != nullptr)
        ConvertNode(pCurrent->m_pRight, pLastNodeInList);
}