46.二叉树展开为链表(学习)
给你二叉树的根结点 root ,请你将它展开为一个单链表:
展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null 。
展开后的单链表应该与二叉树 先序遍历 顺序相同。
示例 1:
输入:root = [1,2,5,3,4,null,6]
输出:[1,null,2,null,3,null,4,null,5,null,6]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [0]
输出:[0]
提示:
树中结点数在范围 [0, 2000] 内
-100 <= Node.val <= 100
解析:
一、递归基:如果当前节点为空(root === null),则直接返回,因为没有什么可以展开的。
二、递归展开左右子树:首先,递归地展开当前节点的左子树和右子树。这是因为在处理当前节点之前,我们需要确保它的所有子树都已经被正确地展开了。
三、处理当前节点:
1.将当前节点的左子树(root.left)暂存到一个临时变量(temp)中。
2.将当前节点的左子指针(root.left)设置为null,因为展开后的单链表中左子指针应该始终为null。
3.将当前节点的右子指针(root.right)设置为原来左子树的根节点(即root.left在步骤1中的值,但此时它已经是null了,因为我们刚刚将其移到了temp中)。
四、找到新右子树的最右节点:
1.从当前节点开始,沿着右子指针(现在指向原来的左子树)一直遍历到最右端。
2.这个最右节点将是新右子树中最后一个节点,我们需要将其右子指针设置为原来右子树的根节点(即temp)。
五、完成展开:现在,当前节点已经按照要求展开了,它的左子指针为null,右子指针指向了一个单链表(可能是空的,如果原来就没有右子树的话),这个单链表包含了原来左子树的所有节点,并且如果原来就有右子树的话,右子树现在被接在了这个链表的末尾。
六、递归继续:由于我们是递归地处理每个节点,所以这个过程会自动地应用到树中的所有节点上,直到整棵树都被展开成一个单链表。
var flatten = function (root) {
if (!root) return; // 如果根节点为空,则直接返回
flatten(root.left); // 递归展开左子树
flatten(root.right); // 递归展开右子树
// 展开操作:将左子树接到右子树的位置,并找到左子树的最右节点,将其右指针指向原来的右子树
let temp = root.right;
root.right = root.left;
root.left = null;
// 找到左子树(现在已经在右边)的最右节点
let p = root;
while (p.right !== null) {
p = p.right;
}
// 将原来的右子树接到最右节点之后
p.right = temp;
};