题目:已知两颗二叉树,将它们合并成一颗二叉树。合并规则是:都存在的结点,就将结点值加起来,否则空的位置就由另一个树的结点来代替。例如:
两颗二叉树是:
Tree 1
1
/ \
3 2
/
5
Tree 2
2
/ \
1 3
\ \
4 7
合并后的树为
3
/ \
4 5
/ \ \
5 4 7
Holy shit!一道简单的题目被我搞得巨复杂。。。。。
先是想到,如果两个子树的节点都为 null ,那么直接 return,然后分情况处理,都不为 null,就加和,创建一个新节点,然后处理左右子树,有一个节点为 null时,根据左子树还是右子树创建相应的节点,然后连到一起。。。。。
然后发现,当两个节点都不为空直接加没错,但是不需要采用一个额外的树来存储,直接合并到 t1 中即可,然后分析当 t2 为 null,太好了,不用合并了,直接 return,当 t1==null 时,也不错, 令 t1=t2即可,然后发现这个函数传参是传引用还是传值的问题,t1 仅仅是数值改变了,其结构并没有改变...
然后标准结局看解析,其实和我的思路也差不多嘛。。。
完整代码如下:
public TreeNode mergeTrees (TreeNode t1, TreeNode t2) {
// write code here
if(t1==null) {
return t2;
}
if(t2==null) {
return t1;
}
t1.val = t1.val+t2.val;
t1.left = mergeTrees(t1.left, t2.left);
t1.right = mergeTrees(t1.right, t2.right);
return t1; //将两棵树合并到t1中
}