题目:
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
注意: 合并必须从两个树的根节点开始。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-two-binary-trees
思路:
- 直接使用先序遍历的递归法,对左右子树进行判断和叠加,这里使用了没有影响原二叉树的递归法。
- 或者使用队列实现的广度优先的迭代法,直接在原二叉树t1上操作,对满足不为空的的左右子结点进行叠加,否则直接将t2点对应结点进行赋值
代码(递归法):
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
if(t1==null || t2==null) return t1==null?t2:t1;
// 先合并根结点
TreeNode root = new TreeNode((t1==null?0:t1.val)+(t2==null?0:t2.val));
// 递归合并左右子树
root.left = mergeTrees(t1==null?null:t1.left, t2==null?null:t2.left);
root.right = mergeTrees(t1==null?null:t1.right, t2==null?null:t2.right);
return root;
}
}
代码(迭代法):
import java.util.LinkedList;
class Solution {
public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
if(t1==null || t2==null) return t1==null?t2:t1;
LinkedList<TreeNode> queue = new LinkedList<>();
queue.add(t1);
queue.add(t2);
while(queue.size()>0){
TreeNode r1 = queue.remove();
TreeNode r2 = queue.remove();
// 直接在t1上操作,合并值
r1.val += r2.val;
// 存入左子树
if(r1.left!=null && r2.left!=null){
queue.add(r1.left);
queue.add(r2.left);
}
// 如果左子树为空直接赋值
if(r1.left==null){
r1.left = r2.left;
}
if(r1.right!=null && r2.right!=null){
queue.add(r1.right);
queue.add(r2.right);
}
if(r1.right==null){
r1.right = r2.right;
}
}
return t1;
}
}