题目描述:有一个搜索二叉树,对于树中的任意一个节点,其左子树的所有值小于该节点的值,而其右子树的所有值大于该节点的值。要求:在不定义任何新的节点情况下,将该二叉树转换成为一个排序双向链表。
假设有上面的二叉树,那么其转换为双向链表后的顺序应该为:2 3 4 5 6 7 8
对于链表头节点:其直接前驱为空指针,直接后继为其父节点;
对于链表尾节点:其直接前驱为其父节点,直接后继为空;
对于左子树的根节点:其直接前驱为其左子树中的最大值;其直接后继为其右子树中的最小值或者父节点的值;
对于右子树的根节点:其直接前驱为其左子树中的最大值或者父节点的值;其直接后继为右子树中的最大值;
因此,我们可以将二叉树看作三个部分:根节点、左子树、右子树。
如果我们可以递归地将左子树和右子树转换成为排序的双向链表,那么整个二叉树也就被转换成了排序的双向链表。
示例代码如下:
#include<iostream>
using namespace std; struct BinTreeNode { int value; BinTreeNode *LeftNode; BinTreeNode *RightNode; }; void ConvertNode(BinTreeNode *pNode, BinTreeNode **pLastNodeInList); BinTreeNode *Convert(BinTreeNode *pRootNode) { BinTreeNode *pLastNodeInList = nullptr; ConvertNode(pRootNode, &pLastNodeInList); BinTreeNode *pHeadOfList = pLastNodeInList; //此时pHeadOfList是二叉树的头节点,要返回链表头需要将它向左移,直到到达链表头
while (pHeadOfList != nullptr&&pHeadOfList->LeftNode != nullptr) { pHeadOfList = pHeadOfList->LeftNode; } //返回链表的头节点
return pHeadOfList; } void ConvertNode(BinTreeNode *pNode, BinTreeNode **pLastNodeInList) { if (pNode == nullptr) { return; } BinTreeNode *pCurrentNode = pNode; //如果存在左子树,那么递归地修改左子树节点指向
if (pCurrentNode->LeftNode != nullptr){ ConvertNode(pNode->LeftNode, pLastNodeInList);} //将当前节点的左子节点指向链表尾部元素(因为链表尾部元素更小)
pCurrentNode->LeftNode = *pLastNodeInList; //将链表尾部元素的右子节点指向当前节点,与上一步操作类似;
if (*pLastNodeInList != nullptr){ (*pLastNodeInList)->RightNode = pCurrentNode; } //更新链表尾部元素
*pLastNodeInList = pCurrentNode; //递归的修改右子树的节点指向
if (pCurrentNode->RightNode != nullptr) { ConvertNode(pCurrentNode->RightNode, pLastNodeInList); } } int main() {}
将上述代码对上面图示的二叉树进行转换,其转换过程中的变量变化如下,这能帮助我们理解整个转换的过程:
在Convert函数中调用ConvertNode()函数结束以后,用来存储链表尾部元素的变量值为二叉树的根节点的值,即:10。
因此,为了正确地返回链表头部元素,需要将该指针向左移到链表头地位置,最终将这个指针返回。