二叉搜索树与双向链表 ----《剑指offer》面试题27

题目

  输入一颗二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。

思路

  可以用分治的思想来解决该问题。如果要将一颗BST(二叉搜索树)转换成一个排序的双向链表，可以将BST简化的看成只有三个结点的树。根结点是BST的根结点，另外两个结点分别代表BST的左子树与又子树。同样又能将左子树与右子树细化成同样的结点。此时，只要懂得如何将三个结点的简单的BST转成排序的双向链表，就能通过递归将一颗大的BST转成排序好的双向链表。
  
  那么，如何将只有三个结点的BST转换成排序好的双向链表？首先，用BinaryTreeNode* pLastNodeInList表示双向链表的尾结点，用来保存输出的结果。此时，左结点已是叶子结点，可以看成一棵已经排序好的双向链表的根结点。那么可以将pLastNodeInList与左结点连接，并更新pLastNodeInList指向双向链表的尾结点。根结点的左子树已经排序好了，利用中序遍历的方法可以将这颗BST的root结点与右子树也排序好。将root结点与已经将左子树排序好的pLastNodeInList连接。再将BST的右子树用同样的方法排序，root结点与右子树已经连接。这样整个BST转换成一个排序的双向链表了，最终pLastNodeInList指向双向链表的最后一个结点。

代码

#include <iostream>

using  namespace std;

struct BinaryTreeNode
{
    int     m_nValue;
    BinaryTreeNode*     m_pLeft;
    BinaryTreeNode*     m_pRight;
};

//  pNode:表示子树的根结点，将些结点加入到双向链表的尾部，成为双向列表的尾结点。
//  pLastNodeInList：双向链表的尾结点，通过连接pNode不断更新。
//                   将其设为指针的指针，方便递归时数据共享。
//  ConvertNode：将BST转换成排序好的双向链表
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;

    //  pNode表示的根结点已经加入到双向链表的尾部，此时，要更新pLastNodeInList的值，
    //  继续让pLastNodeInList指向双向链表的尾结点，并且，此时（*pLastNodeInList）
    //  此时(*pLastNodeInList)->left中已经连接到右子树部分
    *pLastNodeInList = pCurrent;

    //  将右子树也通过同样的方法排序，保存到双向链表中
    if (pCurrent->m_pRight != nullptr)
        ConvertNode(pCurrent->m_pRight, pLastNodeInList);
}


int main()
{
    BinaryTreeNode n1{4, nullptr, nullptr};
    BinaryTreeNode n2{8, nullptr, nullptr};
    BinaryTreeNode n3{12, nullptr, nullptr};
    BinaryTreeNode n4{16, nullptr, nullptr};
    BinaryTreeNode n5{6, &n1, &n2};
    BinaryTreeNode n6{14, &n3, &n4};
    BinaryTreeNode n7{10, &n5, &n6};
    BinaryTreeNode *tail = nullptr;
    ConvertNode(&n7, &tail);

    //  从后向前遍历
    BinaryTreeNode *currentNode = tail;
    while (currentNode)
    {
        cout << currentNode->m_nValue << ' ';
        currentNode = currentNode->m_pLeft;
    }

    return EXIT_SUCCESS;
}