一起学习LeetCode热题100道(46/100)

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;
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值