0617. 合并二叉树【简单】
1. 📝 题目描述
给你两棵二叉树:root1 和 root2。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:
- 如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值
- 否则,不为 null 的节点将直接作为新二叉树的节点
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
示例 1:
txt
输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出:[3,4,5,5,4,null,7]1
2
2
示例 2:
txt
输入:root1 = [1], root2 = [1,2]
输出:[2,2]1
2
2
提示:
- 两棵树中的节点数目在范围
[0, 2000]内 -10^4 <= Node.val <= 10^4
2. 🎯 s.1 - 递归合并
js
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root1
* @param {TreeNode} root2
* @return {TreeNode}
*/
var mergeTrees = function (root1, root2) {
// 如果 root1 为空,返回 root2
if (!root1) return root2
// 如果 root2 为空,返回 root1
if (!root2) return root1
// 两个节点都存在,创建新节点,值为两节点值之和
const merged = new TreeNode(root1.val + root2.val)
// 递归合并左右子树
merged.left = mergeTrees(root1.left, root2.left)
merged.right = mergeTrees(root1.right, root2.right)
return merged
}1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
- 时间复杂度:
,其中 m 和 n 分别是两棵树的节点数,只需遍历较小树的所有节点 - 空间复杂度:
,递归调用栈的深度取决于较小树的高度
算法思路:
- 递归终止条件:任一节点为空时,直接返回另一个节点
- 两节点都存在时,创建新节点,值为两节点值之和
- 递归合并左右子树,分别赋值给新节点的左右子节点
- 返回新创建的合并节点,不修改原树结构
