题目:给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点
示例1:
注意:合并必须从两个树的根节点开始。
解题思路:
从两个树的根节点开始合并,对根节点合并之后还有对根节点的左右子树进行合并,很显然,可以使用递归来解决。
public class demo {
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode() {}
TreeNode(int val) { this.val = val; }
TreeNode(int val, TreeNode left, TreeNode right) {
this.val = val;
this.left = left;
this.right = right;
}
}
public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
//两个二叉树中的节点有一个为空时,合并后为两个二叉树中的非空节点
if(root1==null) {
return root2;
}
if(root2==null) {
return root1;
}
TreeNode merge = new TreeNode(root1.val+root2.val);
//对左右子树进行合并
merge.left = mergeTrees(root1.left,root2.left);
merge.right = mergeTrees(root1.right,root2.right);
return merge;
}