617.合并二叉树

合并二叉树

给你两棵二叉树 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] 内
  • $-10^4$ <= Node.val <= $10^4$

解析

递归合并,同时遍历两棵树。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
/**
* @param {TreeNode} root1
* @param {TreeNode} root2
* @return {TreeNode}
*/
var mergeTrees = function (root1, root2) {
if (!root1) return root2;
if (!root2) return root1;

root1.val += root2.val;
root1.left = mergeTrees(root1.left, root2.left);
root1.right = mergeTrees(root1.right, root2.right);
return root1;
};

简洁的递归解法,直接在 root1 上修改。时间复杂度 O(min(m,n))。


617.合并二叉树
https://leetcode.lz5z.com/617.merge-two-binary-trees/
作者
tickli
发布于
2024年10月31日
许可协议