leetcode:617. 合并二叉树(JavaScript)

617. 合并二叉树

难度简单945收藏分享切换为英文接收动态反馈

给你两棵二叉树: root1root2

想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。

返回合并后的二叉树。

注意: 合并过程必须从两个树的根节点开始。

示例 1:

img

输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出:[3,4,5,5,4,null,7]

示例 2:

输入:root1 = [1], root2 = [1,2]
输出:[2,2]

提示:

  • 两棵树中的节点数目在范围 [0, 2000]
  • -104 <= Node.val <= 104

思路:采用递归,用root1保存合并后的节点,最后返回即可。

要点:递归结束条件

这里需要仔细思考,当root1为空时,有两种情况

  1. root2不为空,以上面图为例:当root1为2节点,root2为3节点处理右子树进入下一层时,此时root2为节点7,root1为空,直接将root2返回给上一层,所以由代码可以看出root1.right(节点2)的值为节点7。注意这里有点绕,要好好想一想递归特性及层级关系

    如果节点7的左子树、右子树不为空呢? 因为节点7保存有它的左右子树的信息,所以当把节点7返回个上一层时,就等于说是吧节点7及后边所有节点都返回给了上一级。

  2. root2为空,还是以上图为例,当root1为2节点,root2为3节点处理左子树进入下一层时,此时root2,root1都为空,将root2返回上一层后,root1.left值为空,符合题意。

// 若root1为空,就返回root2
    if (!root1) return root2;
// 同理
    if (!root2) return root1;

完整代码

var mergeTrees = function(root1, root2) {
    // 若root1为空,就返回root2
    if (!root1) return root2;
    if (!root2) return root1;
    // 若前面的条件都不符合,说明root1,oot2都存在,将root2的值和root2的值相加保存在root1上
    root1.val += root2.val;
    // 递归处理左子树
    root1.left = mergeTrees (root1.left, root2.left);
    // 递归处理右子树
    root1.right = mergeTrees (root1.right, root2.right);
    return root1;
};
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值