携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第26天,点击查看活动详情 >>
一、题目描述 LeetCode - 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]
二、解题思路
本题要求将两个二叉树合并成一棵,两个树相同位置都有节点时将节点的值相加,一个树有而另一棵树没有时将不为空的节点作为新二叉树的节点。本题可以采用二叉树深度优先搜索来完成。对于在搜索过程中出现的每种情况采用不同的合并方式:
- 当两个节点重叠时,将两节点的值相加作为新的二叉树的节点。
- 当其中一个节点为null时,将不为null的节点作为新二叉树的节点。
- 若两个节点均为null,则新二叉树此位置也为null。
三、代码
class Solution {
public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
if (root1 == null){
return root2;
}
if (root2 == null){
return root1;
}
TreeNode m = new TreeNode(root1.val + root2.val);
m.left = mergeTrees(root1.left, root2.left);
m.right = mergeTrees(root1.right, root2.right);
return m;
}
}
四、总结
本题使用深度优先搜索方法,通过遍历两棵二叉树来完成二叉树的合并。由于要递归遍历两个二叉树进行合并,所以时间复杂度为O(min(m,n))其中m,n为两棵二叉树的节点个数。空间复杂度为O(min(m,n))其中m,n分别表示两棵二叉数的层数。