【必拿下系列】leetcode617. 合并二叉树

链接: 题目地址
给你两棵二叉树： root1 和 root2 。

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

返回合并后的二叉树。

注意: 合并过程必须从两个树的根节点开始。
这一题虽然是简单,但是刚接触还是有点不知所措的,(建议大家写代码之前,一定要先思考全面再写,不然可能思路越写越乱)
例子如下:
输入：root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出：[3,4,5,5,4,null,7]
思路如下:

1. 我们要选择把一个树改成最终的目标树(省点空间嘛)
2. 选择是递归还是迭代
3. 递归如下:
4. 退出条件肯定是:if(root1==null)时return root2,(root2==null)return root1;
   得出这个条件还是挺容易的,毕竟两两求和嘛,一个为0,肯定答案就是另一个了.
5. 递归函数直接套用原函数
6. 单层递归逻辑:这又细分为前序/中序/后序遍历,不过也都差不多,具体看代码吧:

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;

    }