题目:输入一颗二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何的新的节点,只能调整树中节点指针的指向。
思路:
首先,我们知道:在二叉树中,每个结点都有两个指向子结点的指针。在双向链表中,每个结点也有两个指针,它们分别指向前一个结点和后一个结点。其次,由于要求转换之后的链表是排好序的,我们可以中序遍历树中的每一个结点,这是因为中序遍历算法的特点是按照从小到大的顺序遍历二叉树的每一个结点。最后,按照中序遍历的顺序,当我们遍历转换到根结点(值为10的结点)时,它的左子树已经转换成一个排序的链表了,并且处在链表中的最后一个结点是当前值最大的结点。我们把值为8的结点和根结点链接起来,此时链表中的最后一个结点就是10了。接着我们去遍历转换右子树,并把根结点和右子树中最小的结点链接起来。很明显,转换它的左子树和右子树,由于遍历和转换过程是一样的,很自然地想到可以用递归。(原先指向左子树的节点指针调整为链表中指向前一个节点的指针,原先指向右子节点的指针调整为链表中指向后一个节点的指针)
struct BinaryTreeNode
{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
};
void ConvertNode(BinaryTreeNode* pRootOfTree, BinaryTreeNode **pLastNodeInList) {
if (NULL == pRootOfTree) {
return;
}
// 转换左子树
if (pRootOfTree->m_pLeft != NULL) {
ConvertNode(pRootOfTree->m_pLeft, pLastNodeInList);
}
// 左子树转换完成,连接根与左子树
pRootOfTree->m_pLeft = *pLastNodeInList; //如果pRootTree->m_pLeft 为空,则指向NULL, 否则指向前一个节点
if (NULL != *pLastNodeInList) {
((*pLastNodeInList)->m_pRight = pRootOfTree); // 前一个节点的m_pRight指针指向当前节点
}
*pLastNodeInList = pRootOfTree; // 将当前节点归入链表末尾(pLastNodeInList 后移)
//转换右子树
if (NULL != pRootOfTree->m_pRight) {
ConvertNode(pRootOfTree->m_pRight, pLastNodeInList);
}
}
BinaryTreeNode *Convert(BinaryTreeNode* pRootOfTree) {
if (NULL == pRootOfTree) {
return NULL;
}
BinaryTreeNode *pLastNodeInList = NULL;
ConvertNode(pRootOfTree, &pLastNodeInList);
// pLastNodeInList指向双向链表的尾节点,需要返回头结点
while (pLastNodeInList != NULL && pLastNodeInList->m_pLeft != NULL) {
pLastNodeInList = pLastNodeInList->m_pLeft;
}
return pLastNodeInList;
}