题目
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示
数据范围:
输入二叉树的节点数 0≤n≤1000,二叉树中每个节点的值 0≤val≤1000
要求:空间复杂度 O(1)(即在原树上操作),时间复杂度 O(n)
注意:
1.要求不能创建任何新的结点,只能调整树中结点指针的指向。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode,有左右指针,其实可以看成一个双向链表的数据结构
4.你不用输出双向链表,程序会根据你的返回值自动打印输出
输入描述:
二叉树的根节点
返回值描述:
双向链表的其中一个头节点。
解析
我们引入一个新的结点pre,由图可知,对于当前结点root,有root->left要指向前继preNode(中序遍历时,对于当前结点root,其左孩子已经遍历完成了,此时root->left可以被修改。);同时,preNode->right要指向当前结点(当前结点是preNode的后继),此时对于preNode结点,它已经完全加入双向链表。
代码
static struct TreeNode *pre; //定义一个树的前驱指针
void Inorder(struct TreeNode *pRoot) //构建中序遍历序列,即左中右的顺序
{
if (!pRoot)
return;
Inorder(pRoot->left); //
pRoot->left = pre; //左子树就是双向链表中的前驱
if (pre)
pre->right = pRoot; //前驱的后继就是当前结点
pre = pRoot; //即向后遍历一个结点
Inorder(pRoot->right);
}
struct TreeNode* Convert(struct TreeNode* pRootOfTree ) {
struct TreeNode *p;
p = pRootOfTree;
if (p)
{
while (p->left)
p = p->left;
pre = p->left;
Inorder(pRootOfTree);
}
return p;
}