算法学习(五)将两颗二叉树进行合并

描述

已知两颗二叉树,将它们合并成一颗二叉树。合并规则是:都存在的结点,就将结点值加起来,否则空的位置就由另一个树的结点来代替。例如:
两颗二叉树是:
                                                                    Tree 1

算法学习(五)将两颗二叉树进行合并_第1张图片



                                                                        Tree 2
 

算法学习(五)将两颗二叉树进行合并_第2张图片


                                                                    合并后的树为

算法学习(五)将两颗二叉树进行合并_第3张图片

数据范围:树上节点数量满足 0 \le n \le 5000≤n≤500,树上节点的值一定在32位整型范围内。

进阶:空间复杂度 O(1)O(1) ,时间复杂度 O(n)O(n)

我的解法:

        以tree1为基础,将tree2上的值或节点加到tree1上,完成合并,递归过程中,当tree1的当前节点为null时,将tree2的节点接在tree1上,当tree2的当前节点为null时,直接返回,都不为空,则将tree1的值加上tree2的值,完成二叉树的合并

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */    
public TreeNode mergeTrees (TreeNode t1, TreeNode t2) {
        // write code here
        if(t1==null) return t2;
        merge(t1, t2);
        return t1;
    }
    public void merge(TreeNode tree1, TreeNode tree2) {
        if(tree1==null) return;//tree1为空,直接在上层将tree2赋值给tree1
        if (tree2 != null) {
            tree1.val += tree2.val;
        }else
            return; //tree2为空不需要在这条路径上递归了,直接返回
        merge(tree1.left, tree2.left);
        tree1.left=tree1.left==null?tree2.left:tree1.left;//赋值
        merge(tree1.right, tree2.right);
        tree1.right=tree1.right==null?tree2.right:tree1.right;//赋值
    }

你可能感兴趣的:(算法学习,1024程序员节)