617. 合并二叉树
难度简单945收藏分享切换为英文接收动态反馈
给你两棵二叉树: root1
和 root2
。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
示例 1:
输入: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为空时,有两种情况
-
root2不为空,以上面图为例:当root1为2节点,root2为3节点处理右子树进入下一层时,此时root2为节点7,root1为空,直接将root2返回给上一层,所以由代码可以看出root1.right(节点2)的值为节点7。注意这里有点绕,要好好想一想递归特性及层级关系。
如果节点7的左子树、右子树不为空呢? 因为节点7保存有它的左右子树的信息,所以当把节点7返回个上一层时,就等于说是吧节点7及后边所有节点都返回给了上一级。
-
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;
};