题目描述
描述:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示:
数据范围:输入二叉树的节点数 0≤n≤1000,二叉树中每个节点的值 0≤val≤1000。
要求:空间复杂度O(1)(即在原树上操作),时间复杂度O(n)。
注意:
1.要求不能创建任何新的结点,只能调整树中结点指针的指向。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继;
2.返回链表中的第一个节点的指针;
3.函数返回的TreeNode,有左右指针,其实可以看成一个双向链表的数据结构;
4.你不用输出双向链表,程序会根据你的返回值自动打印输出;
输入描述:二叉树的根节点。
返回值描述:双向链表的其中一个头节点。
输入:{10,6,14,4,8,12,16}
返回值:From left to right are:4,6,8,10,12,14,16;From right to left are:16,14,12,10,8,6,4;
说明:输入题面图中二叉树,输出的时候将双向链表的头节点返回即可。
输入:{5,4,#,3,#,2,#,1}
返回值:From left to right are:1,2,3,4,5;From right to left are:5,4,3,2,1;
说明:
5
/
4
/
3
/
2
/
1
树的形状如上图
解题思路
二叉搜索树与双向链表:最直观的想法是,中序遍历二叉搜索树,并使用一个数组保存中序遍历的结点指针结果,再遍历中序数组将其左右相连,最后返回中序遍历数组的第一个元素即可。
vector<TreeNode*> TreeList; //中序遍历数组
//中序遍历 得到中序遍历数组
void dfs(TreeNode* cur)
{
if(!cur)
return;
dfs(cur->left);
TreeList.push_back(cur);
dfs(cur->right);
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(!pRootOfTree) //空树
return nullptr;
dfs(pRootOfTree);
for(int i=0;i<TreeList.size()-1;i++) //遍历中序数组 左右相连
{
TreeList[i]->right=TreeList[i+1];
TreeList[i+1]->left=TreeList[i];
}
return TreeList[0];
}
优化:上述方法开辟了额外的数组存储空间,那么有没有什么原地操作的方法呢?当然有!使用二叉树双指针即可!其中使用一个指针pre表示前驱结点,使用一个指针cur表示当前结点,仍然使用中序遍历方法,当左子树遍历完成后,即肯定cur->left为空才返回,此时将cur->left设置为前驱节点,同时,如果pre不为空的话,当前节点必定为前驱节点的后继节点,即pre->right=cur,当然,也要更新前驱节点为当前结点,再继续右子树的遍历,最后返回二叉树的最左下即中序遍历的开头结点即可。(其实就是中序遍历,然后在处理中间结点时,其左边指向前驱,如果前驱不为空,那么前驱的后继就指向当前,这样就避免了再遍历一次来完成后继的填充,其与数组类似,只不过是将数组转化为了指针)
TreeNode *pre=nullptr;
//中序遍历 得到中序遍历数组
void dfs(TreeNode* cur)
{
if(!cur) //叶子结点
return;
dfs(cur->left); //左子树遍历
cur->left=pre; //更新前驱
if(pre) //更新后继
pre->right=cur;
pre=cur; //更新下一次的前驱
dfs(cur->right); //右子树遍历
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(!pRootOfTree) //空树
return nullptr;
dfs(pRootOfTree); //中序遍历
while(pRootOfTree->left) //找到最左下
pRootOfTree=pRootOfTree->left;
return pRootOfTree; //双向链表头结点
}
注意,是通过更新当前的左子树为前驱,同时当前驱不为空,来更新前驱的右子树为当前来实现双向奔赴的啊!!