leetcode刷题：二叉树19（合并二叉树）

617.合并二叉树

力扣题目链接

给定两个二叉树，想象当你将它们中的一个覆盖到另一个上时，两个二叉树的一些节点便会重叠。

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠，那么将他们的值相加作为节点合并后的新值，否则不为 NULL 的节点将直接作为新二叉树的节点。

示例 1:

注意: 合并必须从两个树的根节点开始。

以其中一棵树为基准，递归同时遍历两棵树，每次判断当前结点在非基准树里若存在，则相加，不存在，则跳过，当前结点不存在，在非基准树里存在，则用传来的父结点添加这个结点，记得标记一下是哪棵树。

package com.programmercarl.tree;

import com.programmercarl.util.GenerateTreeNode;

import java.util.ArrayDeque;
import java.util.Deque;

/** * @ClassName MergeTrees * @Descriotion TODO * @Author nitaotao * @Date 2022/7/5 18:02 * @Version 1.0 **/
public class MergeTrees {
    

    public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
    
        if (root1 == null && root2 != null) {
    
            return root2;
        }
        if (root1 != null && root2 == null) {
    
            return root1;
        }
        if (root1 == null && root2 == null) {
    
            return null;
        }
        merge(root1, root2, false,null);
        return root1;
    }

    public void merge(TreeNode root1, TreeNode root2, boolean isLeft,TreeNode parent) {
    
        if (root1 == null && root2 == null) {
    
            return;
        }
        // 以第一棵树为基准
        if (root1 != null && root2 != null) {
    
            root1.val = root1.val + root2.val;
        }

        if (root1 == null && root2 != null) {
    
            //如果 1 没有，2 有
            if (isLeft) {
    
                parent.left = root2;
                return;
            } else {
    
                parent.right = root2;
                return;
            }
        }
        //如果 1 有 ， 2 没有 不做改变
        if (root1 != null && root2 == null) {
    
            return;
        }
        merge(root1.left, root2.left, true,root1);
        merge(root1.right, root2.right, false,root1);
    }

    public static void main(String[] args) {
    
        new MergeTrees().mergeTrees(
                GenerateTreeNode.generateTreeNode("[9,-1,null,-2,0,-4,null,null,8,-5,-3,6,null,null,null,null,null,null,7]")
                ,
                GenerateTreeNode.generateTreeNode("[-1,-2,0,null,null,null,8,6,null,null,7]")
        );
    }

}

来看看大佬的代码

   // 递归
    public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
    
        if (root1 == null) return root2;
        if (root2 == null) return root1;

        TreeNode newRoot = new TreeNode(root1.val + root2.val);
        newRoot.left = mergeTrees(root1.left,root2.left);
        newRoot.right = mergeTrees(root1.right,root2.right);
        return newRoot;
    }