题面描述
题意解析
给定两棵二叉树A和B,若某个节点A和B都有,则合并后的节点为A和B的节点上的数值加和。若某个节点仅有其中一棵有,则直接将其作为合并后新树的节点。
思路分析
递归的进行操作,若其中一棵树的节点为空,则直接返回另一棵树的节点。否则对于节点的值进行加和,可以创建一个新的节点,也可以在原来某棵树的节点上直接进行操作。对于左子节点和右子节点进行递归操作。
代码
/**
* 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) {
TreeNode ans = dfs(root1, root2);
return ans;
}
public TreeNode dfs(TreeNode root1, TreeNode root2){
if(root1 == null) return root2;
if(root2 == null) return root1;
root1.val += root2.val;
root1.left = dfs(root1.left, root2.left);
root1.right = dfs(root1.right, root2.right);
return root1;
}
}