输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示
数据范围:输入二叉树的节点数 0 \le n \le 10000≤n≤1000,二叉树中每个节点的值 0\le val \le 10000≤val≤1000
要求:空间复杂度O(1)O(1)(即在原树上操作),时间复杂度 O(n)O(n)
注意:
1.要求不能创建任何新的结点,只能调整树中结点指针的指向。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode,有左右指针,其实可以看成一个双向链表的数据结构
4.你不用输出双向链表,程序会根据你的返回值自动打印输出
void reconstruct(struct TreeNode* pRootOfTree){
if(pRootOfTree == NULL){
return;
}
if(pRootOfTree->left == NULL && pRootOfTree->right == NULL){
return;
}
struct TreeNode *lo = pRootOfTree->left, *hi=pRootOfTree->right;
if(lo != NULL){
//将左子树看成一串链表,链表最右侧需要与根节点建立双向链表
while(lo->right != NULL){
lo = lo->right;
}
lo->right = pRootOfTree;
pRootOfTree->left = lo;
}
if(hi != NULL){
//将右子树看成一串链表,链表最左侧需要与根节点建立双向链表
while(hi->left != NULL){
hi = hi->left;
}
hi->left = pRootOfTree;
pRootOfTree->right = hi;
}
}
void post_order(struct TreeNode* pRootOfTree){
//进行后续遍历,先将子树转换成双向链表,然后处理根节点
if(pRootOfTree == NULL){
return;
}
post_order(pRootOfTree->left);
post_order(pRootOfTree->right);
reconstruct(pRootOfTree);
}
struct TreeNode* Convert(struct TreeNode* pRootOfTree ) {
if(pRootOfTree == NULL){
return NULL;
}
post_order(pRootOfTree);
struct TreeNode* ret = pRootOfTree;
//使用双向链表找到链表的最左侧节点,并返回
while(ret->left != NULL){
ret = ret->left;
}
return ret;
}
分析:
进行后续遍历递归完成,从树的最底层,将子树转成链表,然后认为每个子树都是链表。其中左子树的最右侧节点与根节点建立链表,右子树的最左侧节点与根节点建立链表。