链接: 题目地址
给你两棵二叉树: root1 和 root2 。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
这一题虽然是简单,但是刚接触还是有点不知所措的,(建议大家写代码之前,一定要先思考全面再写,不然可能思路越写越乱)
例子如下:
输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出:[3,4,5,5,4,null,7]
思路如下:
- 我们要选择把一个树改成最终的目标树(省点空间嘛)
- 选择是递归还是迭代
- 递归如下:
- 退出条件肯定是:if(root1==null)时return root2,(root2==null)return root1;
得出这个条件还是挺容易的,毕竟两两求和嘛,一个为0,肯定答案就是另一个了. - 递归函数直接套用原函数
- 单层递归逻辑:这又细分为前序/中序/后序遍历,不过也都差不多,具体看代码吧:
class Solution {
public TreeNode mergeTrees(TreeNode root1, TreeNode root2){
if(root1==null)return root2;
if(root2==null)return root1;
//后序遍历
//左
root1.left=mergeTrees(root1.left,root2.left);
//右
root1.right=mergeTrees(root1.right,root2.right);
//中
root1.val+= root2.val;
return root1;
}
}
class Solution {
public TreeNode mergeTrees(TreeNode root1, TreeNode root2){
//用中序递归
if(root1==null)return root2;
if(root2==null)return root1;
//左
root1.left=mergeTrees(root1.left,root2.left);
//中
root1.val+= root2.val;
//右
root1.right=mergeTrees(root1.right,root2.right);
return root1;
}
}
迭代的话,需要额外注意:进栈的顺序和什么时候进栈,什么时候不用进栈可以直接处理,当然使用队列也ok
代码如下:
class Solution {
public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
//感觉用前序遍历或者中序遍历或者后序遍历应该都没什么大问题
//那就前序吧(中左右)
//用迭代吧,如果一方节点为空则直接以另一个节点替代
if(root1==null)return root2;
if(root2==null)return root1;
Deque<TreeNode> nodes=new LinkedList<>();
//Deque<TreeNode> nodes2=new LinkedList<>();
//就不用太多而外的空间了,直接在root1上做更改
nodes.push(root2);//1,3
nodes.push(root1);//1,2
while(!nodes.isEmpty()){
TreeNode pop1=nodes.pop();
TreeNode pop2=nodes.pop();
pop1.val+=pop2.val;
/*if(pop1==null&&pop2!=null){
pop1=pop2;//这边发现应该从上一个节点入手
}*/
if(pop1.right!=null&&pop2.right!=null){
nodes.push(pop2.right);
nodes.push(pop1.right);
}
if(pop1.left!=null&&pop2.left!=null){
nodes.push(pop2.left);
nodes.push(pop1.left);
}
if(pop2.left!=null&&pop1.left==null){
pop1.left=pop2.left;
}
if(pop2.right!=null&&pop1.right==null){
pop1.right=pop2.right;
}
}
return root1;
}