给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
注意: 合并必须从两个树的根节点开始。
思路:用递归的方式求解,首先先将root1,root2 的值相加,然后再将root1,root2的左子树,右子树分别合并(递归),返回left和right。然后再新建一个根节点,使根节点的左节点和右节点=left和right,返回合并后的根节点。
递归的终止条件是传入的两个根节点都为null。
还需要注意当root1或root2其中一个为null时,就将该节点的val值看作0。
而且要注意当root1或root2其中一个为null时,将root1,root2的左子树,右子树分别合并(递归)的过程要注意传入的值也要为null,因为root节点为null,其一定没有左右子树。
/**
* Definition for a binary tree node.
* 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;
* }
* }
*/
class Solution {
public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
if(root1==null&&root2==null)
{
return null;
}
int l = (root1==null)?0:root1.val;
int r = (root2==null)?0:root2.val;
TreeNode l1 = (root1==null)?null:root1.left;
TreeNode l2 = (root2==null)?null:root2.left;
TreeNode r1 = (root1==null)?null:root1.right;
TreeNode r2 = (root2==null)?null:root2.right;
TreeNode left = mergeTrees(l1,l2);
TreeNode right = mergeTrees(r1,r2);
return new TreeNode(l+r,left,right);
}
}