一、题目
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
示例 1:
注意: 合并必须从两个树的根节点开始。
二、思路及代码实现
方法一:迭代
考虑将 Tree 2 合并至 Tree 1,利用队列的特性来实现(需要注意:入队时两棵树要保持同步,即左孩子同时入队或者右孩子同时入队),过程如下:
- 给 Tree 1 和 Tree 2 分别创建一个队列,刚开始先将两棵树的根节点入队;
- 只要队列二不空,每次将两个队列的队头出队,将它们的值相加并赋给 Tree 1 对应的节点;
- 考虑两个队头的左孩子——如果两个节点都有左孩子,将两个左孩子分别入队;如果 Tree 2 有左孩子,Tree 1 没有,则给 Tree 1 创建值为 0 的左孩子,然后将两个左孩子分别入队;如果两个节点都没有左孩子,什么都不用做;
- 右孩子同理。
- 最后返回 Tree 1 的根节点。
代码如下:
class Solution {
public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
if(t1 == null)
return t2;
if(t2 == null)
return t1;
LinkedList<TreeNode> queue1 = new LinkedList<>();
LinkedList<TreeNode> queue2 = new LinkedList<>();
queue1.add(t1);
queue2.add(t2);
while(!queue2.isEmpty()){
TreeNode node1 = queue1.poll();
TreeNode node2 = queue2.poll();
node1.val = node1.val + node2.val;
if(node2.left != null){
if(node1.left == null){
node1.left = new TreeNode(0);
}
queue1.add(node1.left);
queue2.add(node2.left);
}
if(node2.right != null){
if(node1.right == null){
node1.right = new TreeNode(0);
}
queue1.add(node1.right);
queue2.add(node2.right);
}
}
return t1;
}
}
方法二:递归(来自官方题解)
我们可以对这两棵树同时进行前序遍历,并将对应的节点进行合并。在遍历时,如果两棵树的当前节点均不为空,我们就将它们的值进行相加,并对它们的左孩子和右孩子进行递归合并;如果其中有一棵树为空,那么我们返回另一颗树作为结果;如果两棵树均为空,此时返回任意一棵树均可(因为都是空)。
代码如下:
public class Solution {
public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
if (t1 == null)
return t2;
if (t2 == null)
return t1;
t1.val += t2.val;
t1.left = mergeTrees(t1.left, t2.left);
t1.right = mergeTrees(t1.right, t2.right);
return t1;
}
}