617. 合并二叉树

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。

示例 1:

输入: 
	Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7                  
输出: 
合并后的树:
	     3
	    / \
	   4   5
	  / \   \ 
	 5   4   7
注意: 合并必须从两个树的根节点开始。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-two-binary-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {

    private TreeNode build(TreeNode root1, TreeNode root2) {
        if (root1 == null && root2 == null) {
            return null;
        }
        TreeNode root;
        if (root1 == null) {
            root = new TreeNode(root2.val);
            root.left = build(root1, root2.left);
            root.right = build(root1, root2.right);
        } else if (root2 == null) {
            root = new TreeNode(root1.val);
            root.left = build(root1.left, root2);
            root.right = build(root1.right, root2);
        } else {
            root = new TreeNode(root1.val + root2.val);
            root.left = build(root1.left, root2.left);
            root.right = build(root1.right, root2.right);
        }
        return root;
    }

    public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        return build(root1, root2);
    }
}


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;
    }
}

posted @ 2021-12-28 14:17  Tianyiya  阅读(37)  评论(0)    收藏  举报