题目:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。如下图,左边的二叉搜索树,则输出转换之后的排序双向链表。
二叉树的结点定义如下:
struct BinaryTreeNode
{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
}
分析:在二叉树中,每一个结点都有两个指向子结点的指针。在双向链表中,每一个结点也有两个指针,它们分别指向前一个结点和后一个结点。由于这两种结点的结构类似,同时二叉搜索树也是一种排序的数据结构,依次在理论可能实现二叉搜索树和排序的双向链表的转换。在二叉搜索树中,左子结点的值总是小于父结点的值,右子结点的值总是大于父结点的值。因此我们在转换排序双向链表时,原先指向左子结点的指针调整为指向前一个结点指针,原先指向右结点的指针调整为指向后一个结点指针。接下来考虑怎么转换。
由于转换之后的链表是排好序的,我们可以中序遍历树中的每一个结点,这是因为中序遍历的特点,按照从小到大的顺序遍历二叉树。当遍历到根结点时,我们把树看成三部分:值为10的结点、根结点值为6的左子树、根结点值为14的右子树。根据排序链表的定义,值为10的结点将和它的左子树的最大一个结点8链接起来,同时它好将和右子树最小的结点12链接起来。
测试用例:功能测试(输入的二叉树是完全二叉树,所有结点都没有左/右子树的二叉树,只有一个结点的二叉树)
特殊输入测试(指向二叉树根结点的指针为NULL指针)
代码如下:
struct BinaryTreeNode
{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
}
BinaryTreeNode* Convert(BinaryTreeNode* pRootofTree)
{
BinaryTreeNode* pLastNodeInList=NULL;
ConvertNode(pRootofTree,&pLastNodeInList);
BinaryTreeNode* pHeadOfList=pLastNodeInList;
while(pHeadOfList !=NULL && pHeadOfList->m_pLeft!=NULL)
pHeadOfList=pHeadOfList->m_pLeft;
return pHeadOfList;
}
void ConvertNode(BinaryTreeNode* pNode,BinaryTreeNode** pLastNodeInList)
{
if(pNode==NULL)
return;
BinaryTreeNode* pCurrent=pNode;
if(pCurrent->m_pLeft!=NULL)
ConvertNode(pCurrent->m_pLeft,pLastNodeInList);
pCurrent->m_pLeft=*pLastNodeInList;
if(*pLastNodeInList!=NULL)
(*pLastNodeInList)->m_pRight=pCurrent;
*pLastNodeInList=pCurrent;
if(pCurrent->m_pRight!=NULL)
ConvertNode(pCurrent->m_pRight,pLastNodeInList);
}