题目:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
解题思路:
根据二叉搜索树的特点,左子树的结点<根节点<右子树的结点,我们可以按照二叉树中序遍历的顺序对该二叉树的指针进行调整,使得二叉树变为一个有序的双向链表。
采用递归的思路,先对左子树进行转换,
再处理根节点,让根节点的做指针指向它的前驱节点;如果其前驱节点非空,让其前驱节点的右指针指向当前结点。注意更新前驱节点;
最后处理右子树,对右子树进行转换。
代码实现:
void TreeToList(TreeNode* pRootOfTree, TreeNode** prev)
{
if(pRootOfTree == NULL)
return;
//先对左子树进行转化
TreeToList(pRootOfTree->left, prev);
//处理当前结点,让当前结点的左指针指向中序遍历的前驱节点,如果前驱节点非空,让前驱节点的右指针指向当前结点
pRootOfTree->left = *prev;
if(*prev)
(*prev)->right = pRootOfTree;
//更新前驱节点
*prev = pRootOfTree;
//在对右子树进行转化
TreeToList(pRootOfTree->right, prev);
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(pRootOfTree == NULL)
return NULL;
//查找链表的头结点
TreeNode* head = pRootOfTree;
while(head->left != NULL)
{
head = head->left;
}
//根据中序遍历的思路进行转换
TreeNode* prev = NULL;
TreeToList(pRootOfTree, &prev);
return head;
}