要求:输入棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的节点,只能调整树种结点指针的指向。
具体见剑指offer题27。
分析:在二叉树中,每个节点都有两个指向子节点的指针。而在双向链表中,每一个节点也有两个指针,分别指向向前和后向的节点。因此这两种数据结构有相似之处,并且二叉搜索树是一种排序的数据结构,可以转换为排序的双向链表。
主要思想:将原先指向左子节点的指针调整为链表中指向前一个节点的指针,原先指向右子节点的指针调整为链表中指向后一个节点的指针。
使用中序遍历二叉搜索树,这样遍历的数据正好是按大小顺序排列的。我们把树看成三部分:值为10的节点,根节点值为6的左子树,和根节点为14的右子树。将左右子树转换为排序的双向链表之后再和根节点链接起来,就能完成题目要求。很明显要用递归。
刚开始看的话还是比较绕的,慢慢看的话会体会到其中的巧妙。
主代码:
BinaryTreeNode* Convert(BinaryTreeNode* pRootOfTree)
{
BinaryTreeNode *PlastNodeInList = NULL;
ConvertNode(pRootOfTree, &pLastNodeInList);
//pLastNodeInList是指向节点指针的指针,它负责指向双链表的尾节点
//返回转换后双向链表头节点
B *pHeadOfList = pLastNodeInList;
while(pHeadOfList != NULL && pHeadOfList->m_pLeft != NULL)
pHeadOfList = pHeadOfList->m_pLeft;
return pHeadOfList;
}
void ConvertNode(B* pNode, B** pLastNodeInList)
{
if( pNode == NULL)
return;
B *pCurrent = pNode;
//这里进行递归,首先是节点10,然后6,都满足if判断语句,所以卡在这,继续调用ConvertNode,到达节点4,节点4左子节点为空,不满足if判断语句,进入下面的代码,开始调整节点指针。最内层递归执行完之后,跳出,回到节点6开始往下执行,此时*pLastNodeInList指向节点4,pCurrent指向节点6,然后对接点4、6建立双向链表联系。后面的类似。
if( pCurrent->m_pLeft != NULL)
ConvertNode(pCurrent->m_pLeft,pLastNodeInList);
//下面三行代码作用:调整二叉树指针方向,二叉树中指向左子节点的指针->指向双向链表中当前节点的前节点;指向右子节点的指针->指向链表当前节点的后节点。
pCurrent->m_pLeft = *pLastNodeInList;
if(*pLastNodeInList != NULL)
(*pLastNodeInList->m_pRight) = pCurrent;
//当前节点也是双向链表的尾节点
*pLastNodeInList = pCurrent;
//遇到右子节点不为空时,执行。首先执行到的是节点“8”
if( pCUrrent->m_pRight != NULL)
ConvertNode(pCurrent->m_pRight,PlastNodeInList);
}