描述
已知两颗二叉树,将它们合并成一颗二叉树。合并规则是:都存在的结点,就将结点值加起来,否则空的位置就由另一个树的结点来代替。例如:
两颗二叉树是:
Tree 1
Tree 2
合并后的树为
数据范围:树上节点数量满足 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;//赋值
}