[Leetcode]617. 合并二叉树

给你两棵二叉树： root1 和 root2 。

想象一下，当你将其中一棵覆盖到另一棵之上时，两棵树上的一些节点将会重叠（而另一些不会）。你需要将这两棵树合并成一棵新二叉树。合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；否则，不为 null 的节点将直接作为新二叉树的节点。

返回合并后的二叉树。

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

示例 1：

输入：root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出：[3,4,5,5,4,null,7]
示例 2：

输入：root1 = [1], root2 = [1,2]
输出：[2,2]
 

提示：

两棵树中的节点数目在范围 [0, 2000] 内
-104 <= Node.val <= 104

code:

public class MergeTrees {

    // 617. 合并二叉树
    public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        if (root1 == null && root2 == null) {
            return null;
        } else if (root1 == null || root2 == null) {
            return root1 == null ? root2 : root1;
        } else {
            TreeNode root = new TreeNode(root1.val + root2.val);
            root.left = mergeTrees(root1.left, root2.left);
            root.right = mergeTrees(root1.right, root2.right);
            return root;
        }
    }
    
    public static void main(String[] args) {
        int[] data1 = new int[]{1,3,2,5};
        int[] data2 = new int[]{2,1,3,-1,4,-1,7};
        TreeNode t1 = creatBTree(data1,0,data1.length);
        TreeNode t2 = creatBTree(data2,0,data2.length);
        MergeTrees solution = new MergeTrees();
        TreeNode t3 = solution.mergeTrees(t1,t2);

        List<List<Integer>> result = solution.levelOrder(t3);
        for(List<Integer> res :result){
            for(Integer x :res){
                System.out.print(x + " ");
            }
            System.out.print('\n');
        }
    }

    public static TreeNode creatBTree(int[] data, int index, int n)
    {
        TreeNode pNode = null;
        if (index < n && data[index]!=-1)
        {
            pNode = new TreeNode(data[index]);
            pNode.left = creatBTree(data, 2 * index + 1, n);  //将二叉树按照层序遍历, 依次标序号, 从0开始
            pNode.right = creatBTree(data, 2 * index + 2, n);
        }
        return pNode;
    }

    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        if(root == null){
            return result;
        }
        LinkedList<TreeNode> que = new LinkedList<TreeNode>();
        que.add(root);
        while(que.size()>0){
            List<Integer> tmpResult = new ArrayList<Integer>();
            int len = que.size();
            for(int i = 0;i<len;i++){
                TreeNode treeNode = que.remove();
                if(treeNode.left !=null){
                    que.add(treeNode.left);
                }
                if(treeNode.right !=null){
                    que.add(treeNode.right);
                }
                tmpResult.add(treeNode.val);
            }
            result.add(tmpResult);
        }
        return result;
    }

}