Leetcode_617. 合并二叉树

题面描述

题意解析

给定两棵二叉树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; 
    }
}