leetcode之二叉树十: 合并二叉树

原题: https://leetcode-cn.com/problems/merge-two-binary-trees/

 

一、题目要求

给定两个二叉树，想象当你将它们中的一个覆盖到另一个上时，两个二叉树的一些节点便会重叠。

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

示例 1:

输入: 
    Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7                  
输出: 
合并后的树:
         3
        / \
       4   5
      / \   \ 
     5   4   7

注意: 合并必须从两个树的根节点开始。

二、解题

package com.leetcode.test.tree;

public class Solution10 {
    public static void main(String[] args) {
        TreeNode a1 = new TreeNode(1);
        TreeNode a2 = new TreeNode(2);
        TreeNode a3 = new TreeNode(3);
        a1.left = a2;
        a1.right = a3;

        TreeNode b1 = new TreeNode(1);
        TreeNode b2 = new TreeNode(2);
        TreeNode b3 = new TreeNode(3);
        b1.left = b2;
        b1.right = b3;

        mergeTrees(a1,b1);

    }

    public static TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        if (root1 == null){
            return root2;
        }
        if (root2 == null){
            return root1;
        }

        TreeNode node = new TreeNode(root1.val + root2.val);        //新的节点
        node.left = mergeTrees(root1.left, root2.left);             //新节点的左节点递归调用
        node.right = mergeTrees(root1.right, root2.right);          //新节点的右节点递归调用

        return node;
    }
}

三、运行结果

 

四、提交结果