描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示
数据范围:输入二叉树的节点数 0≤n≤10000≤n≤1000,二叉树中每个节点的值 0≤val≤10000≤val≤1000
要求:空间复杂度O(1)O(1)(即在原树上操作),时间复杂度 O(n)O(n)
注意:
1.要求不能创建任何新的结点,只能调整树中结点指针的指向。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode,有左右指针,其实可以看成一个双向链表的数据结构
4.你不用输出双向链表,程序会根据你的返回值自动打印输出
输入描述:
二叉树的根节点
返回值描述:
双向链表的其中一个头节点。
示例1
输入:
{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;
说明:
输入题面图中二叉树,输出的时候将双向链表的头节点返回即可。
示例2
输入:
{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 树的形状如上图
解析
1.首先保证有序,那就是中序遍历
2.为了进行连接,由于是单项箭头,所以必须要建立一个prev指针
核心要求:当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继
为了连接prev和cur,prev->right = cur;是在下一层进行的。
prev = cur,修改一级指针,因此prev需要引用传参。
/*
不采用引用传参,真正出问题的prev = cur;是这个语句
这个语句的目的是更新 prev
指针,使其指向当前遍历到的节点 cur
。然而,如果 prev
是通过值传递的(即传递了一个指针的副本),那么这个赋值操作只会改变函数内部 prev
副本的值,而不会影响到原始的 prev
指针。
修改prev->right ,只需要传入right的上一级指针:prev
如果想修改prev,那就得传入prev的上一级指针:prev的引用或者二级指针
*/
代码
class Solution {
public://prev需要在另一层函数中进行修改,所以引用传参。void InorderConvert(TreeNode* cur, TreeNode*& prev){if (cur == nullptr)return;InorderConvert(cur->left, prev);cur->left = prev; //中序遍历到最小节点之后if (prev) //下一层的prev才能链接到上一层的curprev->right = cur;prev = cur;InorderConvert(cur->right, prev);}TreeNode* Convert(TreeNode* pRootOfTree) {TreeNode* cur = pRootOfTree, *prev = nullptr;InorderConvert(cur, prev);TreeNode* phead = pRootOfTree;while (phead && phead->left) //找到头节点(不能为空){phead = phead->left;}return phead;}
};